我正在使用Pytorch 0.4,当它调用binary_cross_entropy丢失时,我在functional.py中得到了FileNotFoundError异常。这是堆栈跟踪:
loss += torch.mean(F.binary_cross_entropy(F.sigmoid(out0), all0) +
文件“/home/local/.../anaconda2/envs/py27/lib/python2.7/site-packages/torch/nn/functional.py”,第1483行,在binary_cross_entropy中 return torch._C._nn.binary_cross_entropy(input,target,weight,size_average,reduce) RuntimeError:未来的cudaEventSynchronize :: wait:设备端断言触发
我知道像https://github.com/pytorch/pytorch/issues/6932这类问题。我在线程中的用户建议的位置添加了修复程序,但是仍然会发生异常。我很乐意提供任何帮助。 最好, S上。