lightning/tests/tests_pytorch/models
Carlos Mocholí 05084e52cc
Simplify the Trainer core logic (#17017)
2023-03-13 13:43:57 +00:00
..
conf copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
__init__.py
test_amp.py Skip flaky ddp-spawn test on windows (#16942) 2023-03-03 15:26:28 +01:00
test_cpu.py Remove `Trainer(multiple_trainloader_mode)` in favor of `CombinedLoader(mode)` (#16800) 2023-02-20 13:32:06 +01:00
test_ddp_fork_amp.py Introduce new precision layout in PL (#16783) 2023-02-17 17:58:14 +01:00
test_fabric_integration.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_gpu.py Make DDP subprocess the default launcher for multi-device (#16780) 2023-02-20 11:20:50 +00:00
test_hooks.py Simplify the Trainer core logic (#17017) 2023-03-13 13:43:57 +00:00
test_hparams.py Move `Trainer._log_hyperparams` to an utility (#16712) 2023-02-10 09:22:56 +01:00
test_onnx.py PL: Test PyTorch 2.0 pre-release on CPU and CUDA (#16764) 2023-02-28 18:38:40 +01:00
test_restore.py Remove `Trainer(multiple_trainloader_mode)` in favor of `CombinedLoader(mode)` (#16800) 2023-02-20 13:32:06 +01:00
test_torchscript.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_tpu.py Introduce new precision layout in PL (#16783) 2023-02-17 17:58:14 +01:00