[DL]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

torch의 버전 문제로 0.5 미만 버전은 loss 추출할 때

loss[0].item()

위의 코드를 사용했었다.

---

loss.item()

이 코드로 수정하면 된다.