Is there pretrained CNN (e.g. ResNet) for CIFAR-10 or CIFAR-100?
yunjey
We dont offer pre-trained resnet with cifar. You might have to train one yourself.
smth
Maybe NIN with CIFAR? :3
skrish13
skrish13
jekbradbury
Oh okay! I’ll see if I can train it. Thanks again!
skrish13
Didn’t we have a repo of links to PyTorch implementation of various papers? (Or I’m confusing that with Chainer’s similar repo :S)
skrish13
jekbradbury
skrish13
prlz77
Sorry for the dumb question, but how do you load a .pytorch file. Is it the same extension as .pth? This doesn’t seem to work:
Ismail_Elezi
prlz77
What would be TheModelClass for your model?
Ismail_Elezi
@jekbradbury Did you manage to train/load the model successfully? I managed to train it, but I am having a sizes do not match error, when I am trying to load it. I am trying to load it with the following code: but getting:
Ismail_Elezi
karlTUM
Yes, I managed to load ResNets that I trained on CIFAR datasets. The code for that is: The parameters for the model and for the net you are loading should agree. For what is worth, the accuracy I got was: with these hyperparameters: Would be interesting to see what happens if I use some more advanced optimizer like Adam. Anyway, in case you don’t have time to train them, I can upload the models today, during the afternoon.
Ismail_Elezi
prlz77
Hi, Then on which dataset all the models are pretrained?
raaj043
they’re all pre-trained on Imagenet-12
smth
Thank you very much. Its helpfulu
raaj043