lightning/tests/tests_pytorch/models
Jirka Borovec 384c203532
ruff: PT some more fixes (#17569)
2023-05-05 08:25:15 +02:00
..
conf copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
__init__.py
test_amp.py ruff: PT some more fixes (#17569) 2023-05-05 08:25:15 +02:00
test_cpu.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04: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 [pre-commit.ci] pre-commit suggestions (#17271) 2023-04-26 21:37:41 +02:00
test_gpu.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_hooks.py ruff: PT some more fixes (#17569) 2023-05-05 08:25:15 +02:00
test_hparams.py ruff: PT some more fixes (#17569) 2023-05-05 08:25:15 +02:00
test_onnx.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_restore.py ruff: PT some more fixes (#17569) 2023-05-05 08:25:15 +02:00
test_torchscript.py [pre-commit.ci] pre-commit suggestions (#17271) 2023-04-26 21:37:41 +02:00
test_tpu.py [TPU] Rename classes to use XLA instead of TPU (#17383) 2023-04-28 12:36:22 -04:00