-2
$\begingroup$

I'm looking for different models (specifically ResNet18/20, ResNet32/34, VGG16, MobileNet and SqueezeNet) and their parameters after training (i.e., .pth file) that were trained on CIFAR-10 or CIFAR-100. I tried looking for them for hours and couldn't find anything. perhaps someone could refer me to a site which have trained models for CIFAR-x?

Thank you.

$\endgroup$
1
$\begingroup$

You can find pretrained models on CIFAR-10 in this GitHub repository.

Also, for fun you can take any backbone trained on ImageNet from TorchVision models. Just replace the classification part from the Linear layer, outputting 1000 classes to 10 or 100. Then you can freeze most of the layers of the network and fit just some bottom layers with the classification head (last nn.Linear).

Modern papers often use CIFAR-10 for transfer learning and you can find the parameters in the paper or discussion on GitHub like here.

$\endgroup$

Not the answer you're looking for? Browse other questions tagged or ask your own question.