lightning/tests/tests_pytorch/models
Adrian Wälchli 079544a902
Rename PrecisionPlugin -> Precision (#18840)
2023-10-30 16:53:13 -04:00
..
conf copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_amp.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_cpu.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_ddp_fork_amp.py Rename PrecisionPlugin -> Precision (#18840) 2023-10-30 16:53:13 -04:00
test_fabric_integration.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_gpu.py [pre-commit.ci] pre-commit suggestions (#18697) 2023-10-03 22:07:21 +02:00
test_hooks.py Rename PrecisionPlugin -> Precision (#18840) 2023-10-30 16:53:13 -04:00
test_hparams.py Add throughput utilities to Fabric and the Trainer (#18848) 2023-10-30 17:10:29 +01:00
test_onnx.py Drop support for PyTorch 1.11 (#18691) 2023-10-04 20:30:44 +02:00
test_restore.py Save ModelCheckpoint's `last.ckpt` as symlink if possible (#18748) 2023-10-11 11:35:13 -04:00
test_torchscript.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_tpu.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00