lightning/tests/models
Nicki Skafte cefc7f7c32
Feature/log computational graph (#3003)
* add methods

* log in trainer

* add tests

* changelog

* fix tests

* fix tests

* fix tests

* fix tests

* fix tests

* fix tests

* fix tests

* text

* added argument

* update tests

* fix styling

* improve testing
2020-08-19 19:08:46 -04:00
..
data ddp fix for trainer.test() + add basic ddp tests (#2997) 2020-08-16 11:19:57 -04:00
__init__.py CI: split tests-examples (#990) 2020-03-25 07:46:27 -04:00
test_amp.py re-trigger build (#2988) 2020-08-15 21:13:00 -04:00
test_cpu.py Feature/log computational graph (#3003) 2020-08-19 19:08:46 -04:00
test_gpu.py ddp fix for trainer.test() + add basic ddp tests (#2997) 2020-08-16 11:19:57 -04:00
test_grad_norm.py clean tests imports (#2834) 2020-08-06 16:58:51 +02:00
test_hooks.py Continue Jeremy's early stopping PR #1504 (#2391) 2020-06-28 21:36:46 -04:00
test_horovod.py clean tests imports (#2834) 2020-08-06 16:58:51 +02:00
test_hparams.py Fix hparams loading for model that accepts *args (#2911) 2020-08-12 09:58:35 -04:00
test_onnx.py Support limit_mode_batches (int) for infinite dataloader (#2840) 2020-08-07 13:02:36 +02:00
test_restore.py Added strict=False for load_from_checkpoint (#2819) 2020-08-13 16:25:43 -04:00
test_sync_batchnorm.py updated sync bn (#2838) 2020-08-06 01:12:11 +02:00
test_tpu.py Bugfix/2956 tpu distrib backend fix (#2959) 2020-08-13 18:57:23 -04:00