lightning/pytorch_lightning/plugins
Kaushik B 2cbdc01256
Fix checkpoint callback & Trainer.test(_) issue for TPUs (#6654)
* Fix checkpoint callback issue for TPUs

* update changelog

* add barrier

* apply code suggestions

* update trainer test

* remove spaces

* fix tpu tests

* Apply suggestions from code review

* add comment

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2021-03-25 10:37:37 +00:00
..
environments introduce default cluster environment for lightning-specific ddp (#5915) 2021-03-05 01:47:29 +00:00
precision Feature/double precision (#6595) 2021-03-24 15:47:58 +05:30
training_type Fix checkpoint callback & Trainer.test(_) issue for TPUs (#6654) 2021-03-25 10:37:37 +00:00
__init__.py Feature/double precision (#6595) 2021-03-24 15:47:58 +05:30
base_plugin.py Add mypy typing to precision plugins. (#6149) 2021-02-26 14:27:16 +01:00