lightning/pytorch_lightning/core
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
..
__init__.py CI: add flake8 (#4239) 2020-10-19 21:20:17 +01:00
datamodule.py Add trainer attribute to datamodule (#3749) 2020-10-01 00:41:19 +05:30
decorators.py
grads.py PyTorch 1.7 Stable support (#3821) 2020-10-30 15:42:14 +00:00
hooks.py [Doc] Fix on_train_batch_end description (#4330) 2020-10-26 20:24:38 +05:30
lightning.py [FEAT] Add lambda closure to manual_optimizer_step (#4618) 2020-11-12 19:22:06 +00:00
memory.py Refactor GPUStatsMonitor to improve training speed (#3257) 2020-09-04 06:02:16 -04:00
saving.py Update old "module_arguments" and "hparams" references in docs (#4417) 2020-11-03 12:13:10 +01:00
step_result.py [FIX] Average Pbar Metrics (#4534) 2020-11-12 15:59:01 +00:00