Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Yuxin Wu's avatar
    additional flop counting using fvcore's flop counter · bc9d5070
    Yuxin Wu authored 
    Summary:
1. save 3 versions of flop count, using both mobile_cv's flop counter and fvcore's flop counter
2. print only a simple short table in terminal, but save others to files

The `print_flops` function seems not used anywhere so this diff just replaced it.

TODO: enable this feature automatically for train/eval workflows in the next diff

Reviewed By: zhanghang1989

Differential Revision: D29182412

fbshipit-source-id: bfa1dfad41b99fcda06b96c4732237b5e753f1bb
    bc9d5070
test_flop_count.py 1.06 KB