Wei_Wong:

How can I save a model like pytorch pretrained model for example mobilenetv2, that I can train my model on this base, save model and use new optimizer or new learning rate like we do on pretrained model?


The work flow would be the same. Once you’ve stored the state_dict of the pretrained model, you could create a new model instance and load the state_dict to it.