본문 바로가기

컴퓨터공학

Boolean value of Tensor with more than one value is ambiguous

반응형

l1 loss를 사용하는 도중 다음과 같은 오류가 발생하였다. 간단히 해결되었다. 

loss = torch.nn.L1Loss(output, pred)

아래와 같이 바꿔준다.

loss = torch.nn.L1Loss()(output, pred)

 

반응형