bryansosa

Train.py for Cifar-10 Error -- RuntimeError: Dimension out of range (

2

I am getting the error described below when runing train.py for cifar-10. Note that I previously run make_datasets.py ` $ python train.py --dataset=cifar10 Epoch 0: 0%| | 0/3125 [00:00