今天在使用pytorch跑模型时遇到了这么一个问题,主要的报错信息包括:
RuntimeError: reduce failed to synchronize: cudaErrorAssert: device-side assert triggered
Assertion `input >= 0. && input <= 1.` failed.
xiaoxifei在他的博客中解释了这个问题,链接: link.
我在这就不赘述原理,给大家说说我遇到的一个具体情况,以及我的解决方法,给大家提供一些参考。
我所做的工作是一个三分类问题,但是在选择损失函数的时候选择了BCELoss,我把损失函数换成CrossEntropyLoss,这个报错就消失了。