RuntimeError: Expected a 'cuda' device type for generator but found 'cpu'
I am trying to train PeleeNet pytorch and got the following error

train.py line 80

pelee_voc train configuration


Turning the shuffle parameter off in the dataloader solved it.
Got the answer form here.
