RuntimeError: reduce failed to synchronize: cudaErrorAssert: device-side assert triggered

  今天在使用pytorch跑模型时遇到了这么一个问题,主要的报错信息包括:

RuntimeError: reduce failed to synchronize: cudaErrorAssert: device-side assert triggered
 Assertion `input >= 0. && input <= 1.` failed.

  xiaoxifei在他的博客中解释了这个问题,链接: link.
  我在这就不赘述原理,给大家说说我遇到的一个具体情况,以及我的解决方法,给大家提供一些参考。
  我所做的工作是一个三分类问题,但是在选择损失函数的时候选择了BCELoss,我把损失函数换成CrossEntropyLoss,这个报错就消失了。

你可能感兴趣的:(深度学习基础,python,python,深度学习)