Cifar exception in tensor size at beginning of training
david-klindt opened this issue · 1 comments
david-klindt commented
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
jhjacobsen commented
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.