strict parameter in model.set_classy_state

facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification

https://classyvision.ai

MIT License

1.59k stars 277 forks source link

strict parameter in model.set_classy_state #619

Closed simran2905 closed 4 years ago

simran2905 commented 4 years ago

Summary: Allow to load partial head states from state_dict. This is useful when we have new parameters in a head and we want to load weights for the existing parameters.

Reviewed By: mannatsingh

Differential Revision: D22133179

facebook-github-bot commented 4 years ago

This pull request was exported from Phabricator. Differential Revision: D22133179

facebook-github-bot commented 4 years ago

This pull request has been merged in facebookresearch/ClassyVision@82b3ac16ed385a64fd7f2099d44bed193e97d9c7.