lightning/pytorch_lightning/core
Jirka Borovec a6e7aa7796
allow using apex with any PT version (#2865)
* wip

* setup

* type

* name

* wip

* docs

* imports

* fix if

* fix if

* use_amp

* Apply suggestions from code review

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* Apply suggestions from code review

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* fix tests

* Apply suggestions from code review

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

* fix tests

* todos

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2020-08-08 11:07:32 +02:00
..
__init__.py Fix docs typo (#2803) 2020-08-03 14:55:17 +02:00
datamodule.py Call DataModule hooks implicitly in trainer (#2755) 2020-08-01 20:17:57 -04:00
decorators.py remove deprecated in v0.9 (#2760) 2020-07-30 23:19:28 +02:00
grads.py Rework of Sklearn Metrics (#1327) 2020-06-10 15:43:12 +02:00
hooks.py allow using apex with any PT version (#2865) 2020-08-08 11:07:32 +02:00
lightning.py Support limit_mode_batches (int) for infinite dataloader (#2840) 2020-08-07 13:02:36 +02:00
memory.py allow using apex with any PT version (#2865) 2020-08-08 11:07:32 +02:00
saving.py clean imports (#2867) 2020-08-08 00:33:51 +02:00
step_result.py clean imports (#2867) 2020-08-08 00:33:51 +02:00