pytorch/torch/optim
Alexander Grund 5b0f400488 Replace list(map(...)) constructs by list comprehensions (#46461)
Summary:
As discussed in https://github.com/pytorch/pytorch/issues/46392 this makes the code more readable and possibly more performant.

It also fixes a bug detected by this where the argument order of `map` was confused: 030a24906e (diff-5bb26bd3a23ee3bb540aeadcc0385df2a4e48de39f87ed9ea76b21990738fe98L1537-R1537)

Fixes https://github.com/pytorch/pytorch/issues/46392

Pull Request resolved: https://github.com/pytorch/pytorch/pull/46461

Reviewed By: ailzhang

Differential Revision: D24367015

Pulled By: ezyang

fbshipit-source-id: d55a67933cc22346b00544c9671f09982ad920e7
2020-10-19 18:42:49 -07:00
..
_multi_tensor Refactor scalar list APIs to use overloads (#45673) 2020-10-19 09:28:49 -07:00
__init__.py enable torch.optim.swa_utils.SWALR (#42574) 2020-08-05 12:37:45 -07:00
__init__.pyi enable torch.optim.swa_utils.SWALR (#42574) 2020-08-05 12:37:45 -07:00
adadelta.py
adadelta.pyi
adagrad.py [optimizer] introduce optimizer functional API, refactor Adagrad (#44715) 2020-09-25 17:10:26 -07:00
adagrad.pyi
adam.py [optimizer] refactor Adam to use functional API (#44791) 2020-09-25 17:13:08 -07:00
adam.pyi
adamax.py Use amax/maximum instead of max in optimizers (#43797) 2020-09-15 10:39:40 -07:00
adamax.pyi
adamw.py Use amax/maximum instead of max in optimizers (#43797) 2020-09-15 10:39:40 -07:00
adamw.pyi
asgd.py Fix HTTP links in documentation to HTTPS (#40878) 2020-07-06 20:05:21 -07:00
asgd.pyi
functional.py [dist_optim] introduce distributed functional optimizer (#45221) 2020-09-25 17:13:10 -07:00
lbfgs.py Avoid zero division in _cubic_interpolate (#42093) 2020-07-28 08:32:00 -07:00
lbfgs.pyi
lr_scheduler.py Replace list(map(...)) constructs by list comprehensions (#46461) 2020-10-19 18:42:49 -07:00
lr_scheduler.pyi Fix type annotation for CosineAnnealingLR (#41866) 2020-07-23 15:56:50 -07:00
optimizer.py Reference amp tutorial (recipe) from core amp docs (#44725) 2020-09-16 11:37:58 -07:00
optimizer.pyi Add reset_grad() function (#44423) 2020-09-09 22:05:45 -07:00
rmsprop.py Fix HTTP links in documentation to HTTPS (#40878) 2020-07-06 20:05:21 -07:00
rmsprop.pyi
rprop.py
rprop.pyi
sgd.py
sgd.pyi
sparse_adam.py Check SparseAdam params are dense on init (#41966) (#43668) 2020-09-01 14:25:59 -07:00
sparse_adam.pyi
swa_utils.py typo fixes (#41632) 2020-07-20 07:23:00 -07:00
swa_utils.pyi Add SWA to PyTorch mainline (#35032) 2020-04-27 07:42:19 -07:00