search

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

ClassyVision Changes to Support checkpoint_load_strict Flags #757

Closed kailee880209 closed 3 years ago

kailee880209 commented 3 years ago

Summary: It is very hard to extract features from the intermediate layers of a classy model because checkpoint_load_strict is hardcoded to True. This diff allows for setting that to false when user only constructs part of the model while loading the full model state dict to get intermediate layer features.

The flag is default to True so it will not affect existing workflows.

Differential Revision: D29948213

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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