Hi everyone 
I have a pretrained model that I would like to use on a testset. I do the following to check if the model did actually load the weights:
checkpoint = torch.load(checkpoint_path, map_location=device) 
model = Model()
print(model.conv1.weight[0]) # (1) see the weights of the random initialisation

model.load_state_dict(checkpoint['model_state'], strict=False)
print(checkpoint['model_state']['module.conv1.weight'][0]) # (2) see the weights of the pretrained model
print(model.conv1.weight[0]) # I would expect this to be the same as (2) but it is the same as (1)

self.model.eval()

What am I doing wrong?
Any help is very much appreciated!
All the best,
snowe