[Error] IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item<T>()` in C++ to convert a 0-dim tensor to a number

IndexError로 다음과 같은 에러가 발생했다.

 

깃허브에 있는 예전 코드를 가져올 때 발생할 수 있는 에러인데 pyTorch의 버전이 0.5 보다 크면 발생할 수 있다고 한다.

 

이 에러는 단순히 0 차원 텐서가 유효하지 않기 때문에 발생하는 에러다.

data[0] --> data 로 바꿔주면 해결된다.

 

주석처럼 돼있는걸 주석 아래줄로 바꾸면 된다.

 

참고 사이트 : 

github.com/NVIDIA/flownet2-pytorch/issues/113