lightning/pytorch_lightning
Justus Schock 069ae27cef
Accelerator Refactor: Precision Plugins (#5718)
* add basic accelerator class.
Co-Authored with @awaelchi

* add basic trainign type plugin.
Co-Authored with @awaelchi

* pep8

Co-authored-by: @awaelchi

* update copyright

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

* add apex_amp

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

* add mixed base class

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

* add native amp

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

* add native amp sharded

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

* add tpu bfloat

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

* add inits

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

* Update precision_plugin.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-01-31 13:12:02 -05:00
..
accelerators Accelerator Refactor: Precision Plugins (#5718) 2021-01-31 13:12:02 -05:00
callbacks [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
cluster_environments Refactor: legacy accelerators and plugins (#5645) 2021-01-26 20:04:36 -05:00
core [feat] 1/2 Add trainer.predict (#5579) 2021-01-27 11:38:14 -05:00
distributed fix/enable - check F401 (#5201) 2020-12-21 10:15:04 +01:00
loggers docs cleaning - testcode (#5595) 2021-01-26 04:44:54 -05:00
metrics [Metrics] AUC/AUROC class interface (#5479) 2021-01-27 08:16:54 -05:00
overrides Refactor LightningDataParallel (#5670) 2021-01-31 06:08:16 -05:00
plugins Accelerator Refactor: Precision Plugins (#5718) 2021-01-31 13:12:02 -05:00
profiler [feat] Add PyTorch Profiler. (#5560) 2021-01-26 06:48:54 -05:00
trainer Refactor: drop LoggerStages (#5673) 2021-01-30 15:59:18 +01:00
tuner Fix isort a few failures (#5504) 2021-01-15 17:44:27 -05:00
utilities Refactor LightningDataParallel (#5670) 2021-01-31 06:08:16 -05:00
__init__.py update nightly & upgrade Twine (#5458) 2021-01-26 14:29:47 +01:00
py.typed make PyTorch Lightning PEP 561 Compliant (#3187) 2020-09-09 13:37:03 -04:00
setup_tools.py formatting to PL utils (#5713) 2021-01-30 15:28:59 +01:00