I mean I want to use, let’s say - resnet50 as a transfer learning but using the above method, they are asking me to define the model.
I used the below statement -
model_transfer = model.load_state_dict(torch.load('model.path'))