I think I’m misunderstanding something here… If I do the following, will I still be able to use the pretrained parameters of model.Model()?
net = model.Model()
net.load_state_dict(checkpoint["net"])
for c in net.child ... #freeze

model = nn.Sequential(model.Model(), Classifier())