ClassyVision icon indicating copy to clipboard operation
ClassyVision copied to clipboard

Model state should support PyTorch API

Open mannatsingh opened this issue 5 years ago • 2 comments

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

mannatsingh avatar Mar 24 '21 03:03 mannatsingh

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

facebook-github-bot avatar Mar 24 '21 03:03 facebook-github-bot

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

facebook-github-bot avatar Mar 24 '21 04:03 facebook-github-bot