lightning/tests/tests_pytorch/models
awaelchli bf25167bbf
Add testing for PyTorch 2.4 (Trainer) (#20010)
2024-07-11 06:52:56 -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 test: replacing `tmpdir` with `tmp_path` in `tests_pytorch/models` (#19647) 2024-03-17 01:10:05 -04:00
test_cpu.py test: replacing `tmpdir` with `tmp_path` in `tests_pytorch/models` (#19647) 2024-03-17 01:10:05 -04:00
test_ddp_fork_amp.py Rename PrecisionPlugin -> Precision (#18840) 2023-10-30 16:53:13 -04:00
test_fabric_integration.py [pre-commit.ci] pre-commit suggestions (#19229) 2024-01-10 13:11:48 -05:00
test_gpu.py test: replacing `tmpdir` with `tmp_path` in `tests_pytorch/models` (#19647) 2024-03-17 01:10:05 -04:00
test_hooks.py Disable skipping training step in distributed training (#19918) 2024-05-30 11:54:48 -04:00
test_hparams.py Set `weights_only` in tests to avoid warnings in PyTorch 2.4 (#20057) 2024-07-08 04:38:27 -04:00
test_onnx.py Support pathlib.Path file paths when saving ONNX models (#19727) 2024-04-03 20:42:25 -04:00
test_restore.py Set `weights_only` in tests to avoid warnings in PyTorch 2.4 (#20057) 2024-07-08 04:38:27 -04:00
test_torchscript.py Add testing for PyTorch 2.4 (Trainer) (#20010) 2024-07-11 06:52:56 -04:00
test_tpu.py test: replacing `tmpdir` with `tmp_path` in `tests_pytorch/models` (#19647) 2024-03-17 01:10:05 -04:00