lightning/pytorch_lightning/plugins
chaton 6bc4490d01
[HotFix] Resolve TPU Training (#6027)
* fix tpus

* update

* add back reduction in val_loss

* resolve some bugs with TPUs

* update changelog

* update on comments

* forgot status

* Fix train_bn arg

* resolve comments

* update on comments

Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
2021-02-17 16:40:13 +00:00
..
environments remove legacy plugins (#5950) 2021-02-16 19:20:58 +00:00
precision [accelerator][FeatBugFix] Improve manual optimization API (#5771) 2021-02-16 16:00:35 -05:00
training_type [HotFix] Resolve TPU Training (#6027) 2021-02-17 16:40:13 +00:00
__init__.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00
base_plugin.py Add PredictLoop (#5752) 2021-02-16 17:11:56 -05:00