flop count: capture output tensors from all layers so that unused layers are correctly counted.

facebookresearch / fvcore

Collection of common code that's shared among different research projects in FAIR computer vision team.

Apache License 2.0

2.03k stars 229 forks source link

Open ppwwyyxx opened 2 years ago

ppwwyyxx commented 2 years ago

Reviewed By: ericmintun

Differential Revision: D32242109

facebook-github-bot commented 2 years ago

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