Hi,
I have a pretrained model and I want to load just part of it in my model as Im replacing the classification layer. What Im doing is:
model.load_state_dict(state_dict, strict=False)
I need just a heads up that this is the correct way to go. Thank you very much.