Seonglyong Gong f80ef73d1c [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		2022-09-28 02:48:07 +00:00
..
_multi_tensor
__init__.py
__init__.pyi
_functional.py	Adding maximize to SparseAdam (#80336)	2022-07-08 12:17:27 +00:00
adadelta.py	Change docstring type callable to Callable for consistency (#82487)	2022-08-01 17:26:09 +00:00
adadelta.pyi
adagrad.py	Change docstring type callable to Callable for consistency (#82487)	2022-08-01 17:26:09 +00:00
adagrad.pyi
adam.py	Revert "resubmit: [mta] APEX style Fused Adam (#81705) (#85507)"	2022-09-27 16:59:35 +00:00
adam.pyi
adamax.py	Handle complex optimization in Adamax by treating complex numbers as 2D real numbers (#80319)	2022-08-05 21:03:18 +00:00
adamax.pyi
adamw.py	Handle complex optimization in AdamW by treating complex numbers as 2D real numbers (#80280)	2022-08-05 13:47:14 +00:00
adamw.pyi
asgd.py	[optim] asgd : handle complex params as independent real params (#84472)	2022-09-06 16:58:42 +00:00
asgd.pyi
lbfgs.py	Change docstring type callable to Callable for consistency (#82487)	2022-08-01 17:26:09 +00:00
lbfgs.pyi
lr_scheduler.py	CyclicLR memory leak fix (#85462)	2022-09-27 17:41:58 +00:00
lr_scheduler.pyi	CyclicLR memory leak fix (#85462)	2022-09-27 17:41:58 +00:00
nadam.py	Change docstring type callable to Callable for consistency (#82487)	2022-08-01 17:26:09 +00:00
nadam.pyi
optimizer.py	[Profiler] tracking Optimizer (part 2 of Record Optimizer) (#84920)	2022-09-28 02:48:07 +00:00
optimizer.pyi	Fix Optimizer.zero_grad type annotation (#76998)	2022-05-11 00:05:26 +00:00
radam.py	Change docstring type callable to Callable for consistency (#82487)	2022-08-01 17:26:09 +00:00
radam.pyi
rmsprop.py	[optim] rmsprop: handle complex params as independent real params (#83860)	2022-08-22 21:55:01 +00:00
rmsprop.pyi
rprop.py	[optim] rprop: handle complex params as independent real params (#83858)	2022-08-23 08:39:35 +00:00
rprop.pyi
sgd.py	Make sure that we can load old optimizer checkpoint (#83588)	2022-08-17 15:08:05 +00:00
sgd.pyi
sparse_adam.py	Change docstring type callable to Callable for consistency (#82487)	2022-08-01 17:26:09 +00:00
sparse_adam.pyi
swa_utils.py	Integrate xdoctest - Rebased (#82797)	2022-08-12 02:08:01 +00:00
swa_utils.pyi