pytorch
aa63c0d9 - [PyPer] Skip printing out per node time when do_profile is on (#63256)

Commit
3 years ago
[PyPer] Skip printing out per node time when do_profile is on (#63256) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63256 This suppresses printing out the per node time which is very long when the net has too many ops. It can be easily turned on by setting `--pt_sr_print_per_node_time=1`. Reviewed By: ajyu, mikeiovine Differential Revision: D30298331 fbshipit-source-id: 32b3f93b3fe19d335654168311fda93331a1e706
Author
Hao Lu
Parents
Loading