lightning/tests/models
Aki Nitta 9da78a94bd
Rename `TPUSpawnPlugin` to `TPUSpawnStrategy` (#11190)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-12-21 16:36:16 +00:00
..
conf
data
__init__.py
test_amp.py
test_cpu.py Disable validation completely when `overfit_batches>0` (#9709) 2021-12-01 13:57:57 +00:00
test_gpu.py Remove references to torchtext.legacy from PyTorch Lightning (#10724) 2021-11-30 19:32:07 +00:00
test_grad_norm.py
test_hooks.py 3/n Simplify spawn plugins: Merge `pre_dispatch` and `setup` logic (#11137) 2021-12-20 17:41:22 +01:00
test_horovod.py Skip strategy=ddp_spawn, accelerator=cpu, python>=3.9 tests (#10550) 2021-11-16 10:06:47 +05:30
test_hparams.py Enable logging hparams only if there are any (#11105) 2021-12-17 19:40:56 +01:00
test_onnx.py Avoid the deprecated `onnx.export(example_outputs=...)` in torch 1.10 (#11116) 2021-12-17 10:11:11 +01:00
test_restore.py Add required states for resumed ModelCheckpoint GC (#10995) 2021-12-20 17:05:15 +01:00
test_sync_batchnorm.py Teardown sync-batchnorm after training (#11078) 2021-12-16 18:58:44 +00:00
test_torchscript.py
test_tpu.py Rename `TPUSpawnPlugin` to `TPUSpawnStrategy` (#11190) 2021-12-21 16:36:16 +00:00