Using vision.models with the CIFAR dataset?

I’ve looked around but haven’t found much, and have found that

This makes me think think vision#140 and vision#155 are relevant.

Is there any way to use the model implemented in vision.models with CIFAR?

The pre-trained models are all based on Imagenet-12. Most of the model definitions also need 224x224 as the minimum input size. For CIFAR, you are better off with other repos.

Thanks. I’ve got another repo working now:

1 Like