carpedm20/ENAS-pytorch

Cifar10 CNN problem

aaron851113 opened this issue · 1 comments

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)

Has the cnn module could been run?