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