pytorch
f80ef73d - [Profiler] tracking Optimizer (part 2 of Record Optimizer) (#84920)

Commit
2 years ago
[Profiler] tracking Optimizer (part 2 of Record Optimizer) (#84920) Summary: Part 2 of Record Optimizer param_groups and states (https://github.com/pytorch/pytorch/pull/84063) - hooking from optimizer step - PyOptCall Type - declare data type for collection - python binding - simple unit test case Test Plan: buck run mode/opt //caffe2/test:profiler Differential Revision: D39402667 Pull Request resolved: https://github.com/pytorch/pytorch/pull/84920 Approved by: https://github.com/robieta
Author
Committer
Parents
Loading