使用pytorch框架训练神经网络时,可能会遇到如下错误,或者类似的错误:
/pytorch/aten/src/ATen/native/cuda/Loss.cu:106: ... Assertion 'input_var >= zero && input_var < one' failed.
...
...device-side assert triggered
遇到过两次都与pytorch内置的损失函数有关:
1. 类别错误——分类网络输出类别维度与target的范围不一致
2. 传入loss的网络输出结果数值不在指定范围内——nn.BCELoss()的输入记得先使用torch.sigmoid()