lightning/tests/tests_pytorch/models
Adrian Wälchli 3d5c3d24f9
Remove unused auto_collect_arguments class method (#14015)
2022-08-05 08:49:00 +00:00
..
conf
data
__init__.py
test_amp.py
test_cpu.py
test_gpu.py Add support for DDP fork (#13405) 2022-07-22 16:05:35 +00:00
test_grad_norm.py
test_hooks.py
test_horovod.py Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00
test_hparams.py Remove unused auto_collect_arguments class method (#14015) 2022-08-05 08:49:00 +00: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
test_tpu.py Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00