Pytorch BCEloss中的 RuntimeError: reduce failed to synchronize: device-side assert triggered
tensorflow中写代码麻烦,pytorch写起来简单,但是有发现bug,因为是框架内的问题就无法修改。
在确定output已经使用了sigmoid,确定了label和logits都是[0,1]
BCEloss 在运行时还是可能会爆出题目中的错误。
目前我发现比较合理的解决方案是github
可能是因为optimizer是实现的时候把eps加在了logits或者label上,导致了超了[0,1],设定eps为0就好了
如果optimizer的eps是fixed的,可以直接在label中的positive值减去对应的eps值即可。
最新update
最好用 BCEwithlogitloss