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

出现这个错误:

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

在模型最后加上sigmoid() 或者softmax()

你可能感兴趣的:(pytorch)