lightning/tests/tests_pytorch/models
Carlos Mocholí 713a2f79ee
Mark AMP test as flaky (#15055)
2022-10-10 23:26:13 +02:00
..
conf
data Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
__init__.py
test_amp.py Mark AMP test as flaky (#15055) 2022-10-10 23:26:13 +02:00
test_cpu.py Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
test_ddp_fork_amp.py Support ddp_fork strategy with native AMP by attempting NVML-based CUDA availability assessment (#14984) 2022-10-05 18:52:06 -04:00
test_gpu.py Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
test_grad_norm.py Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
test_hooks.py Remove deprecated callback hooks (#14834) 2022-10-10 15:46:28 +00:00
test_horovod.py Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
test_hparams.py Integrate `lightning_utilities.core.imports` (#14475) 2022-09-06 12:56:20 +00:00
test_onnx.py Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
test_restore.py Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00
test_torchscript.py Make internal torchscript check a class attribute (#14904) 2022-09-29 13:40:25 +00:00
test_tpu.py Remove old testing artifacts (#15052) 2022-10-10 17:34:18 +00:00