[파이토치 (PyTorch)] IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

 

0.5 이전 버전의 파이토치에서 작성된 코드를 0.5 이후 버전에서 돌리다 보면 위와 같은 에러가 날 때가 있습니다. 파이토치의 데이터 자료 구조가 바뀌어서 그렇습니다.

 

코드에서 변수.data[0] 인 부분을 변수.data 로 바꾸어주시면 해결됩니다.

 

참고한 곳: https://github.com/NVIDIA/flownet2-pytorch/issues/113#issuecomment-450802359

  1. 영상을압축하자 2022.01.13 18:22 신고

    감사합니다 덕분에 문제를 해결하였습니다.

  2. khojun611 2022.01.17 14:29

    감사합니다.. 덕분에 해결됐네요

  3. 연습중 2022.02.09 14:27

    너무 감사드립니다. a.i 관련 깃헙소스로 테스트하고 있는데
    ai도 모르고 파이썬도 잘 몰라서 헤맸는데 덕분에 해결되었어요

    • 크롬망간 2022.02.12 01:30 신고

      도움이 되었다니 기쁘네요 ^^ 댓글 남겨주셔서 감사합니다 ^^

+ Recent posts