We should only load model state when test_only is True

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

We should only load model state when test_only is True #736

Closed mannatsingh closed 3 years ago

mannatsingh commented 3 years ago

Summary: There were a lot of bugs in evaluation because we were loading state which we didn't need to. The checkpoint state needed unnecessary keys as well which this fixes.

Differential Revision: D27886445

facebook-github-bot commented 3 years ago

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