lightning/pytorch_lightning/trainer
Lezwon Castelino cfd06a083b
Bugfix/2956 tpu distrib backend fix (#2959)
* override dist backend when using tpus

* added test

* updated doc string

* drop redundant info...

* more redundant info

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: rohitgr7 <rohitgr1998@gmail.com>
2020-08-13 18:57:23 -04:00
..
__init__.py pep 8 (#2967) 2020-08-13 18:56:02 -04:00
auto_mix_precision.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
callback_config.py clean imports (#2867) 2020-08-08 00:33:51 +02:00
callback_hook.py updated hooks (#2850) 2020-08-07 09:29:57 -04:00
configuration_validator.py Fix enforce_datamodule_dataloader_override() for iterable datasets (#2957) 2020-08-13 23:06:17 +02:00
data_loading.py allow using apex with any PT version (#2865) 2020-08-08 11:07:32 +02:00
deprecated_api.py remove deprecated in v0.9 (#2760) 2020-07-30 23:19:28 +02:00
distrib_data_parallel.py Bugfix/2956 tpu distrib backend fix (#2959) 2020-08-13 18:57:23 -04:00
distrib_parts.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
evaluation_loop.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
ignored_warnings.py Support returning python scalars in DP (#1935) 2020-08-07 09:18:29 +02:00
logging.py clean imports (#2867) 2020-08-08 00:33:51 +02:00
lr_finder.py clean imports (#2867) 2020-08-08 00:33:51 +02:00
model_hooks.py Datamodule (#2668) 2020-07-24 11:42:15 -04:00
optimizers.py refactor 1/n for v1.0.0 (#2704) 2020-07-25 14:38:51 -04:00
states.py Add tracking of basic states in Trainer [wip - to-be-merged after v0.9] (#2541) 2020-08-09 06:24:09 -04:00
supporters.py Structured results (train loop only. val loop separate PR) (PR 2/5) (#2615) 2020-07-20 19:00:20 -04:00
trainer.py Bugfix/2956 tpu distrib backend fix (#2959) 2020-08-13 18:57:23 -04:00
training_io.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
training_loop.py add apex test (#2921) 2020-08-13 10:03:13 -04:00
training_tricks.py add apex test (#2921) 2020-08-13 10:03:13 -04:00