一种 NaN 导致的错误:
RuntimeError: cudaEventSynchronize in future::wait: device-side assert triggered
这个错误是由于在使用 BCELoss 两分类交叉熵损失时,预测标签的输入需要在 0 - 1 范围之内,而 NaN 的出现则产生改错误。
在网络训练中出现 NaN 的可能原因入下: