Solved by kshitij in post #9 
              
              
                import torchvision.models as models
import torch.nn as nn
model_path = 'Path_to_model/resnet_50.pth'
model_transfer = models.resnet50()
missing_keys, unexpected_keys = model_transfer.load_state_dict(torch.load(model_path))
print(model_transfer)

https://pytorch.org/docs/stable/nn.html?highlight=load…