lightning/tests/tests_pytorch/models
Rohit Gupta 61ae35c378
Use sklearn in runif (#15426)
* Use sklearn in runif
* test by removing sklearn dep
* remove repeated code
* seed
2022-11-01 11:40:32 +00:00
..
conf Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
data/horovod Single source for the mypy version (#15224) 2022-10-21 16:15:33 +09:00
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_amp.py Validate SRUN variables when launching in SLURM (#15011) 2022-10-19 21:42:11 +00:00
test_cpu.py Use sklearn in runif (#15426) 2022-11-01 11:40:32 +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 Use sklearn in runif (#15426) 2022-11-01 11:40:32 +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 Use sklearn in runif (#15426) 2022-11-01 11:40:32 +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 Use sklearn in runif (#15426) 2022-11-01 11:40:32 +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