lightning/tests/tests_pytorch/models
Adrian Wälchli 49caddde6e
Call `configure_model()` in `LM.load_from_checkpoint()` (#19036)
2023-11-21 09:44:18 -05:00
..
conf copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
__init__.py
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 Call `configure_model()` in `LM.load_from_checkpoint()` (#19036) 2023-11-21 09:44:18 -05: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 Fix `rank_zero_only` rank not set in ddp-spawn based strategies (#19030) 2023-11-20 10:49:14 -05:00
test_tpu.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00