lightning/tests/tests_pytorch/models
Adrian Wälchli 28e18881a9
Mark stage argument in hooks as required (#14064)
Co-authored-by: rohitgr7 <rohitgr1998@gmail.com>
2022-09-01 15:47:40 +02:00
..
conf Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
data Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_amp.py Raise an error when resuming training with Apex (#14341) 2022-08-26 13:11:24 +00:00
test_cpu.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_gpu.py Remove support for the deprecated torchtext legacy (#14375) 2022-08-26 20:01:51 +00:00
test_grad_norm.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_hooks.py Enable `on_before_batch_transfer` for `DPStrategy` and `IPUAccelerator` (#14023) 2022-08-18 12:12:29 +00:00
test_horovod.py Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00
test_hparams.py Mark stage argument in hooks as required (#14064) 2022-09-01 15:47:40 +02:00
test_onnx.py add testing PT 1.12 (#13386) 2022-07-15 19:41:23 +02:00
test_restore.py Use `global_step` while restoring logging step for old checkpoints (#13645) 2022-07-19 18:53:22 +00:00
test_torchscript.py Merge pull request #13123 from Lightning-AI/mps_accelerator 2022-06-24 08:15:48 -04:00
test_tpu.py Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00