lightning/tests/tests_pytorch/helpers
Jirka Borovec d9817aa5f1
refactor: move RunIf to PL pkg (#16923)
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-03-03 15:45:59 +00:00
..
__init__.py tests: switch imports for pytorch (#16595) 2023-02-02 10:06:45 +00:00
advanced_models.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
dataloaders.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
datamodules.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
datasets.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
deterministic_model.py Remove the `*_step_end` hooks (#16791) 2023-02-20 13:04:40 +00:00
pipelines.py Update torchmetrics requirement from <0.10.1,>=0.7.0 to >=0.7.0,<0.11.1 in /requirements (#15904) 2023-03-02 23:35:31 +00:00
runif.py refactor: move RunIf to PL pkg (#16923) 2023-03-03 15:45:59 +00:00
simple_models.py Update torchmetrics requirement from <0.10.1,>=0.7.0 to >=0.7.0,<0.11.1 in /requirements (#15904) 2023-03-02 23:35:31 +00:00
test_datasets.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_models.py PL: Test PyTorch 2.0 pre-release on CPU and CUDA (#16764) 2023-02-28 18:38:40 +01:00
utils.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00