반응형
l1 loss를 사용하는 도중 다음과 같은 오류가 발생하였다. 간단히 해결되었다.
loss = torch.nn.L1Loss(output, pred)
아래와 같이 바꿔준다.
loss = torch.nn.L1Loss()(output, pred)
반응형
'컴퓨터공학' 카테고리의 다른 글
Perceiver IO: Text Classification (0) | 2022.07.10 |
---|---|
Perceiver IO: a scalable, fully-attentional model that works on any modality (0) | 2022.07.10 |
[Pytorch] 텐서보드 초간단 사용법 (0) | 2022.06.07 |
초간단 git 사용법 (0) | 2022.05.27 |
[Pytorch] RNN 예제 (0) | 2022.05.25 |