lightning/tests/tests_pytorch
Adrian Wälchli a0ca2c8bcd
Disable memory sharing on model parameters in ddp-spawn (#18238)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-08-15 14:39:51 +02:00
..
accelerators docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
callbacks docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
checkpointing docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
core Refresh the internal LightningOptimizer state for inspection (#18280) 2023-08-11 18:20:09 +02:00
deprecated_api Support true 16-bit precision with FSDP in Trainer (#18219) 2023-08-10 04:15:35 -04:00
graveyard [TPU] Proper half-precision implementation for XLA (#18213) 2023-08-11 11:37:41 -04:00
helpers docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
loggers docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
loops Explicitly enable grad in closure (#18268) 2023-08-10 16:58:29 -04:00
models [TPU] Proper half-precision implementation for XLA (#18213) 2023-08-11 11:37:41 -04:00
overrides Simplify double precision input and module conversion (#18209) 2023-08-01 19:37:15 -04:00
plugins Accommodate FSDP full-precision `param_dtype` training with PyTorch < 2.0 (#18278) 2023-08-14 12:22:26 +02:00
profilers docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
serve Address test flakiness (#18223) 2023-08-03 15:13:55 -04:00
strategies Disable memory sharing on model parameters in ddp-spawn (#18238) 2023-08-15 14:39:51 +02:00
trainer [TPU] Proper half-precision implementation for XLA (#18213) 2023-08-11 11:37:41 -04:00
tuner docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
utilities docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
__init__.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
conftest.py docformatter: config with black (#18064) 2023-08-09 10:44:20 -04:00
run_standalone_tasks.sh copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
run_standalone_tests.sh ci: add current & total count to standalone test output (#17511) 2023-04-28 11:13:53 +02:00
run_tpu_tests.sh Fix [TPU] tests (#18140) 2023-07-24 15:13:36 +02:00
test_cli.py Update jsonargparse[signatures] requirement from <4.23.0,>=4.18.0 to >=4.18.0,<4.24.0 in /requirements (#18307) 2023-08-14 13:49:09 +02:00