Add FLOPs computation to ViT

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

Closed mannatsingh closed 3 years ago

mannatsingh commented 3 years ago

Summary: Also fix flops for dropout

Differential Revision: D28337180

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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