pytorch
d7766937 - [Profiler] Optimizer param_groups (part 3 of Record Optimizer) (#85784)

Commit
2 years ago
[Profiler] Optimizer param_groups (part 3 of Record Optimizer) (#85784) Summary: - use TensorMetadata struct - check_and_store util as overloading - param_groups - clean up unit test cases Test Plan: buck run mode/opt //caffe2/test:profiler Reviewed By: chaekit Differential Revision: D39406072 Pull Request resolved: https://github.com/pytorch/pytorch/pull/85784 Approved by: https://github.com/aaronenyeshi, https://github.com/robieta
Author
Committer
Parents
Loading