Model state should support PyTorch API

facebookresearch / ClassyVision

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

https://classyvision.ai

MIT License

1.59k stars 278 forks source link

Model state should support PyTorch API #727

Open mannatsingh opened 3 years ago

mannatsingh commented 3 years ago

Summary: Classy Models should work like regular PyTorch models. The {get, set}_classy_state functions for state are the only blockers which this diff fixes by moving over to state_dict and load_state_dict

Differential Revision: D25213283

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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