Cifar10 CNN problem
aaron851113 opened this issue · 1 comments
aaron851113 commented
utils.py", line 150, nbatch = data.size(0) // bsz
AttributeError: 'DataLoader' object has no attribute 'size'
-> nbatch = len (data.size)
image.py line 9,
if args == 'cifar' -> if args == 'data/cifar'
line 30-31/38-39,
-> batch_size = 200, shuffle=True,
-> num_workers= 4 , pin_memory=True)
songxueXS commented
Has the cnn module could been run?