기록하는삶

[에러 해결/Pytorch] Assertion `input_val >= zero && input_val <= one` failed 본문

회고, 기록/에러 해결

[에러 해결/Pytorch] Assertion `input_val >= zero && input_val <= one` failed

mingchin 2022. 5. 8. 04:48
728x90
반응형

모델 train 과정에서 발생했다. input에 nan이 포함돼있거나, loss function이 binary cross entropy인데 target으로 설정된 값이 [0,1] 외의 값이 존재하는 등의 문제가 있을 수 있다. 나는 전자의 경우임을 확인, nan을 제거해 해결했다.

728x90
반응형