Open 2hip3ng opened 5 years ago
Is the batch_size=1 when training?
Is the batch_size=1 when training?Thank you!
yes,ImageDataLoader default is 1
so if it is due to the dataset is too small ,so the batchsize should be 1
Is the batch_size=1 when training?