dawn-bench-models
dawn-bench-models copied to clipboard
RuntimeError: dimension specified as 0 but tensor has no dimensions
Hi,
I am having bellow issue running with Pytorch 0.5:
/Users/coredump/DL/dawn-bench-models/pytorch/CIFAR10/benchmark/cifar10/train.py:113: UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number
losses.update(loss.data[0], batch_size)
/Users/coredump/DL/dawn-bench-models/pytorch/CIFAR10/benchmark/cifar10/train.py:76: UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number
tops = list(map(lambda k: corrects.data[0][k - 1], top))
Traceback (most recent call last):
File "/Users/coredump/anaconda3/bin/cifar10", line 11, in
Any ideas?
Rgds, FM
Is this project abandoned? Is there any update related to same issue after 3 years?
I am still having the same problem, please see https://github.com/stanford-futuredata/dawn-bench-models/issues/7