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

Add support to normalize inputs in the heads #664

Closed mannatsingh closed 3 years ago

mannatsingh commented 3 years ago

Summary: Added a normalize_inputs argument to FullyConnectedHead and VisionTransformerHead.

Added simple test cases for both heads

Differential Revision: D25213285

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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