Assertion `input_val >= zero && input_val <= one` failed

Assertion `input_val >= zero && input_val <= one` failed_第1张图片

损失函数输入错误
RuntimeError: CUDA error: device-side assert triggered
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
例如,BCELoss 要求其输入介于 0 和 1 之间。如果输入(模型的输出)超出该特定损失函数的可接受范围,则会触发错误。

MSE Sigmod()

你可能感兴趣的:(算法,python,开发语言)