lightning/tests/models
Jirka Borovec 31a36f04df
define distributed as a type (#3740)
* define type

* miss

* Apply suggestions from code review

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>

* miss

* warn

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
2020-09-30 08:33:01 -04:00
..
data fix lib paths after Wandb 0.10 (#3520) 2020-09-17 08:37:49 -04:00
__init__.py CI: split tests-examples (#990) 2020-03-25 07:46:27 -04:00
test_amp.py disable optimizers setup during testing (#3059) 2020-09-29 01:09:04 +02:00
test_cpu.py [WIP] ref: deprecated results obj, added support for simpler comms (1/n) (#3681) 2020-09-27 23:19:46 -04:00
test_gpu.py implement fix and test (#3459) 2020-09-11 10:55:58 -04:00
test_grad_norm.py fix gradient norm tracking for row_log_interval > 1 (#3489) 2020-09-15 18:41:27 +02:00
test_hooks.py ref: device to gpus (#3405) 2020-09-08 22:14:17 -04:00
test_horovod.py Fix global step increment on training_epoch_end (#3673) 2020-09-27 20:19:51 -04:00
test_hparams.py warn user when dropping unpicklable hparams (#2874) 2020-08-28 09:07:43 +02:00
test_onnx.py fix ONNX model save on GPU (#3145) 2020-08-26 16:22:19 +00:00
test_restore.py enable None model checkpoint default (#3669) 2020-09-26 23:14:04 -04:00
test_sync_batchnorm.py updated sync bn (#2838) 2020-08-06 01:12:11 +02:00
test_torchscript.py to_torchscript method for LightningModule (#3258) 2020-09-03 20:24:44 +02:00
test_tpu.py define distributed as a type (#3740) 2020-09-30 08:33:01 -04:00