lightning/pytorch_lightning
chaton 4018237c30
[FEAT] Add lambda closure to manual_optimizer_step (#4618)
* added lambda_closure

* move to types

* add 2 new tests

* make example more complex

* add complex example to doc

* added more tests

* resolve doc

* typo

* update

* update tpu optimizer_step

* Apply suggestions from code review

* Update pytorch_lightning/core/lightning.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* update

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2020-11-12 19:22:06 +00:00
..
accelerators [FEAT] Add lambda closure to manual_optimizer_step (#4618) 2020-11-12 19:22:06 +00:00
callbacks Replace a MisconfigurationException with warning in ModelCheckpoint callback (#4560) 2020-11-10 10:44:43 +01:00
cluster_environments ref: unify slurm and TE under backendPlugin 1/n (#4578) 2020-11-08 14:28:55 -05:00
core [FEAT] Add lambda closure to manual_optimizer_step (#4618) 2020-11-12 19:22:06 +00:00
distributed notices (#4118) 2020-10-13 07:18:07 -04:00
loggers fix logged keys in mlflow logger (#4412) 2020-11-10 17:20:25 +05:30
metrics Metric ddp bugfix (#4482) 2020-11-10 09:16:31 +01:00
overrides removed support for EvalResult and TrainResult (#3968) 2020-10-07 22:39:16 -04:00
plugins Sharded Accelerator 1/n: Expose clip gradients to plugins via abstract class (#4639) 2020-11-12 17:18:09 +00:00
profiler feature: Allow str arguments in Trainer.profiler (#3656) 2020-10-27 16:27:16 +05:30
trainer [FEAT] Add lambda closure to manual_optimizer_step (#4618) 2020-11-12 19:22:06 +00:00
tuner Skip tuner algorithms on fast dev (#3903) 2020-11-10 00:34:42 +01:00
utilities [Fix] Move log value to cpu. (#4592) 2020-11-10 21:13:41 +00:00
__init__.py Update __init__.py 2020-10-27 18:15:26 -04:00
py.typed make PyTorch Lightning PEP 561 Compliant (#3187) 2020-09-09 13:37:03 -04:00