lightning/tests/tests_pytorch/utilities
awaelchli 34a34a0754
Enable saving and loading stateful DataLoaders in Trainer (#19361)
2024-01-31 21:11:19 -05:00
..
migration ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
testing Compose RunIf utilities (#17520) 2023-05-05 01:21:58 +02:00
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_all_gather_grad.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_auto_restart.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_combined_loader.py Enable saving and loading stateful DataLoaders in Trainer (#19361) 2024-01-31 21:11:19 -05:00
test_compile.py precommit: drop Black in favor of Ruff (#19380) 2024-01-31 17:09:39 +00:00
test_consolidate_checkpoint.py Utility to consolidate sharded checkpoints (#19213) 2024-01-23 17:15:22 -05:00
test_data.py [pre-commit.ci] pre-commit suggestions (#19229) 2024-01-10 13:11:48 -05:00
test_deepspeed_collate_checkpoint.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_deepspeed_model_summary.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_dtype_device_mixin.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_enums.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_exceptions.py ci: testing with py3.11 (#17822) 2023-08-28 15:15:12 +00:00
test_grads.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_imports.py Rename PrecisionPlugin -> Precision (#18840) 2023-10-30 16:53:13 -04:00
test_memory.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_model_helpers.py `_restricted_classmethod`: add wrapper, to allow inspection (#19332) 2024-01-23 18:23:06 -05:00
test_model_summary.py Remember the eval mode of submodules when switching trainer stages (#18951) 2023-11-16 16:32:27 -05:00
test_parameter_tying.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_parsing.py Add AttributeDict container for Fabric (#18943) 2023-11-18 09:25:26 -05:00
test_pytree.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_seed.py Request `torch.cuda` RNG states only if CUDA is available (#19234) 2024-01-10 16:16:29 -05:00
test_signature_utils.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_types.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_upgrade_checkpoint.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_warnings.py Utility to disable all instances of `PossibleUserWarning` (#18744) 2023-10-10 06:53:32 -04:00