컴퓨터공학
Boolean value of Tensor with more than one value is ambiguous
해변에서 코딩하기
2022. 7. 5. 20:06
반응형
l1 loss를 사용하는 도중 다음과 같은 오류가 발생하였다. 간단히 해결되었다.
loss = torch.nn.L1Loss(output, pred)
아래와 같이 바꿔준다.
loss = torch.nn.L1Loss()(output, pred)
반응형