dhwiii's notepad | 딥 러닝, 코덱 일기장

[파이썬 에러코드] IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number 본문

■ Working Draft/◎ 에러코드 일기장

[파이썬 에러코드] IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

dhwiii 2021. 4. 14. 22:03

파이썬 파이토치 에러입니다.

IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

파이썬 파이토치의 버전이 0.5 이전 버전을 사용하게 되면 해당 에러가 나타나게 되는데,

 

 [변경 전] generator_adversarial_loss.data[0]  -> [변경 후] generator_adversarial_loss

이런 식으로 제일 뒤에 있는 [0]을 지워주시면 해결 됩니다.

참고 : github.com/NVIDIA/flownet2-pytorch/issues/113

Comments