meijieru bd054e7cef reduce memory usage for centered rmsprop (#24170) ... Summary: Reduce gpu memory usage by using in-place operation Pull Request resolved: https://github.com/pytorch/pytorch/pull/24170 Differential Revision: D16784495 Pulled By: vincentqb fbshipit-source-id: 03820cdc9a3952b95b9af0f87d3a9bb0f21e9b4d		2019-08-13 12:18:31 -07:00
..
__init__.py	Implement AdamW optimizer (#21250)	2019-07-02 09:09:10 -07:00
__init__.pyi	More type stubs (#18511)	2019-04-01 16:03:58 -07:00
adadelta.py	Added parameter range checks for all optimizers (#6000)	2018-03-28 11:22:23 +02:00
adagrad.py	Revert D14577575: [pytorch][PR] Fix lack of state init for adagrad and add share_memory flag	2019-04-26 15:43:04 -07:00
adam.py	Adam/AdamW implementation minor fix (#22628)	2019-08-01 11:42:04 -07:00
adam.pyi	More type stubs (#18511)	2019-04-01 16:03:58 -07:00
adamax.py	Added parameter range checks for all optimizers (#6000)	2018-03-28 11:22:23 +02:00
adamw.py	Adam/AdamW implementation minor fix (#22628)	2019-08-01 11:42:04 -07:00
asgd.py	Added parameter range checks for all optimizers (#6000)	2018-03-28 11:22:23 +02:00
lbfgs.py	Use lower case for strong wolfe option. (#22092)	2019-06-26 08:20:25 -07:00
lr_scheduler.py	Update CosineAnnealingWarmRestarts to follow PyTorch 1.1+ Step Order. (#23833)	2019-08-07 07:15:50 -07:00
lr_scheduler.pyi	Add interfaces in lr_scheduler.pyi (#23934)	2019-08-12 07:03:41 -07:00
optimizer.py	Lightweight at-most-once logging for API usage (#20745)	2019-05-23 23:17:59 -07:00
optimizer.pyi	Fix optimizer type hint (#20648)	2019-05-22 11:27:40 -07:00
rmsprop.py	reduce memory usage for centered rmsprop (#24170)	2019-08-13 12:18:31 -07:00
rprop.py	Turn on F401: Unused import warning. (#18598)	2019-03-30 09:01:17 -07:00
sgd.py	Updated SGD docs with subscripts (#23985)	2019-08-09 10:32:40 -07:00
sgd.pyi	More type stubs (#18511)	2019-04-01 16:03:58 -07:00
sparse_adam.py	fix lint after new flake8 release added new style constraints (#13047)	2018-10-24 09:03:38 -07:00