lightning/tests/tests_pytorch
Carlos Mocholí 67459944ea
Avoid FSDP deprecations during save/load with newer torch versions (#19463)
* Avoid FSDP deprecations during save/load with newer torch versions

* Refactor

* Tests
2024-02-14 19:43:59 +01:00
..
accelerators Drop XLA XRT support (#19232) 2024-01-10 18:39:20 +01:00
callbacks Support TQDM_MINITERS env variable (#19381) 2024-01-31 20:17:35 -05:00
checkpointing Fix saving relative symlink for ModelCheckpoint callback (#19303) 2024-01-20 09:32:08 -05:00
core Create barrier without timeout in `prepare_data()` (#19448) 2024-02-13 12:10:07 +01:00
demos Improve handling the positional encoding in Transformer example (#18987) 2023-11-19 14:37:31 +01:00
deprecated_api Drop support for PyTorch 1.12 (#19300) 2024-01-26 11:44:24 -05:00
graveyard Remove the Graphcore IPU integration (#19405) 2024-02-12 16:16:02 -05:00
helpers [pre-commit.ci] pre-commit suggestions (#19229) 2024-01-10 13:11:48 -05:00
loggers Fix `CSVLogger` trying to append to file from previous run in same version folder (#19446) 2024-02-13 13:59:04 -05:00
loops Enable saving and loading stateful DataLoaders in Trainer (#19361) 2024-01-31 21:11:19 -05:00
models Create barrier without timeout in `prepare_data()` (#19448) 2024-02-13 12:10:07 +01:00
overrides ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
plugins Delay `Precision.convert_module` until `configure_model` has run (#19061) 2024-02-07 16:27:19 -05:00
profilers Drop nvtx test (#19154) 2023-12-14 15:43:23 +01:00
serve CI: enable testing with coming PT 2.2 (#19289) 2024-01-26 16:42:09 +01:00
strategies Avoid FSDP deprecations during save/load with newer torch versions (#19463) 2024-02-14 19:43:59 +01:00
trainer Remove the Graphcore IPU integration (#19405) 2024-02-12 16:16:02 -05:00
tuner Bugfix/18394 batch size finder max val batches (#18854) 2023-10-25 15:03:42 -04:00
utilities Enable saving and loading stateful DataLoaders in Trainer (#19361) 2024-01-31 21:11:19 -05:00
__init__.py Ignore pytest cleanup warning (#19164) 2023-12-29 07:52:19 +01:00
conftest.py Remove the Graphcore IPU integration (#19405) 2024-02-12 16:16:02 -05:00
run_standalone_tasks.sh Drop nvtx test (#19154) 2023-12-14 15:43:23 +01:00
run_tpu_tests.sh Drop XLA XRT support (#19232) 2024-01-10 18:39:20 +01:00
test_cli.py Update CLI tests to no longer require 3rd party logger dependencies (#18899) 2023-10-31 09:22:17 -04:00