플로라도의 data workout
파이토치 에러 핸들링 - TypeError: cross_entropy_loss(): argument 'input' (position 1) must be Tensor, not NoneType 본문
기초 노트/PyTorch
파이토치 에러 핸들링 - TypeError: cross_entropy_loss(): argument 'input' (position 1) must be Tensor, not NoneType
플로라도 2024. 4. 14. 22:59loss = nn.CrossEntropyLoss()(preds, label.to(device))
에러 메세지 : TypeError: cross_entropy_loss(): argument 'input' (position 1) must be Tensor, not NoneType
학습 루프안의 loss를 계산하는 과정에서 첫번째 포지션 인자인 preds가 NoneType이라는 에러로, 모델의 forward 메서드에서 리턴값이 누락된 경우