lightning/tests/tests_pytorch/models
awaelchli 3518f9e092
Delay DeepSpeed config setup (#19209)
2023-12-24 17:04:04 -05: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 Delay DeepSpeed config setup (#19209) 2023-12-24 17:04:04 -05:00
test_hparams.py Handle more of the flaky tests (#19193) 2023-12-21 14:04:24 +01:00
test_onnx.py Drop support for PyTorch 1.11 (#18691) 2023-10-04 20:30:44 +02:00
test_restore.py Fix ModelCheckpoint tests from incomplete PR (#19205) 2023-12-22 08:49:09 +01: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