You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
another issue is in self.train_data = self.train_data.reshape((45000, 3, 32, 32)),
which result in ValueError: cannot reshape array of size 153600000 into shape (45000,3,32,32)
(1)You can modify models in torchvision.model_zoo to build your own models.
(2)153600000=50000332*32. You should retain 10% of the CIFAR-10 training data (i.e. 5000 images) as the validation set. I split the data_batch_5 into train and val dataset, and it works for me.
HI,
when I run PENCIL.py, there was an import error said no module named 'models',
look forward to see your complete code.
The text was updated successfully, but these errors were encountered: