Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Load pretrained Imagenet models by pytorch

See original GitHub issue

When I load a pretrained imagenet model by pytorch using a finetune task, an AssertionError: Checkpoint does not contain classy_state_dict was raised. So I want to know, how to load a imagenet model to initialize the backbone of a class model? Thank you.

Issue Analytics

State:
Created 4 years ago
Comments:11 (5 by maintainers)

Top GitHub Comments

1reaction

CaiyuZhangcommented, Apr 27, 2020

@mannatsingh Yeah, I thought about this reason too, but I actually update classy_vision before running my code. But this error still happened, I will check out the version, no worry, I have loaded my trained models through other methods, thanks for ur reply.

1reaction

P4ppenheimercommented, Apr 9, 2020

Hello @mannatsingh, thanks for reacting so quickly to our request. Best, Simon

Top Results From Across the Web

Models and pre-trained weights — Torchvision main ... - PyTorch

TorchVision offers pre-trained weights for every provided architecture, using the PyTorch torch.hub . Instancing a pre-trained model will download its weights ...

Image Classification using Pre-trained Models in PyTorch

In this post, we will cover how we can use TorchVision module to load pre-trained models and carry out model inference to classify...

pretrained-models.pytorch - Model Zoo

To load a pretrained models from imagenet: model_name = 'nasnetalarge' # could be fbresnet152 or inceptionresnetv2 model = pretrainedmodels.

PyTorch Pretrained Model - Python Guides

PyTorch pretrained model load · A pretrained model is defined as a neural network model trained on a suitable dataset like AlexNet, ImageNet,...

PyTorch image classification with pre-trained networks

Specifying the pretrained=True flag instructs PyTorch to not only load the model architecture definition, but also download the pre-trained ...