jhjacobsen/pytorch-i-revnet

Cifar exception in tensor size at beginning of training

david-klindt opened this issue · 1 comments

In utils_cifar.py line 105 (and 133):
train_loss += loss.data[0]

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

Yes, the repo is based on an earlier pytorch version, would be great if you can do a pull request if you have made it run with a recent version.