Hello!
This probably means that the state dictionary you have saved in your checkpoint does not contain the weights you’re trying to load. In other words, your checkpoint is not correct, or a different model configuration has been saved. To see what you have actually saved do:
model_state_dict = torch.load(/path/to/your/checkpoint, map_location = lambda storage: loc: storage) saved_weights = model_state_dict.keys() print(saved_weights)
This will tell you what you actually save. It is possible to load those weights in a model that contains only some of the weights in your model_state_dictionary (see here 91)