lightning/tests/tests_pytorch/utilities
Carlos Mocholí b97b3ac053
Avoid `inference_mode` with `torch.compile` (#17215)
2023-03-29 19:43:28 +00:00
..
migration Add a migration for the dataloader loops (#17125) 2023-03-20 17:12:58 +00:00
testing refactor: move RunIf to PL pkg (#16923) 2023-03-03 15:45:59 +00:00
__init__.py
test_all_gather_grad.py Remove memory-retaining epoch-end hooks (#16520) 2023-02-06 17:00:36 +00:00
test_auto_restart.py Sequential `CombinedLoader` to flatten the eval and predict loops (#16726) 2023-02-17 17:37:11 +01:00
test_combined_loader.py Set a fixed stage in the evaluation loops (#17094) 2023-03-27 10:00:03 +02:00
test_compile.py Avoid `inference_mode` with `torch.compile` (#17215) 2023-03-29 19:43:28 +00:00
test_data.py Refactor `loop.setup_data` with utility functions (#16918) 2023-03-09 00:04:42 +00:00
test_deepspeed_collate_checkpoint.py Introduce new precision layout in PL (#16783) 2023-02-17 17:58:14 +01:00
test_deepspeed_model_summary.py Introduce new precision layout in PL (#16783) 2023-02-17 17:58:14 +01:00
test_distributed.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_dtype_device_mixin.py Remove DP (#16748) 2023-02-16 05:12:08 +00:00
test_enums.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_exceptions.py Add `Strategy.on_exception` (#16646) 2023-02-08 14:00:31 +00:00
test_grads.py Log gradient norms of any magnitude (#16877) 2023-02-28 01:32:07 +00:00
test_imports.py update for neptune 1.0 (#16761) 2023-03-21 19:00:07 +00:00
test_memory.py Remove unused code (#17001) 2023-03-09 00:07:48 +00:00
test_model_helpers.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_model_summary.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_parameter_tying.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_parsing.py Backwards compatibility for `get_init_args` (#16851) 2023-02-28 01:09:30 +01:00
test_pytree.py Add more pytree tests (#16825) 2023-02-21 17:27:00 +01:00
test_seed.py tests: switch imports for pytorch (#16595) 2023-02-02 10:06:45 +00:00
test_torchdistx.py Merge DDPStrategy and DDPSpawnStrategy in PL (#16809) 2023-02-27 14:43:23 -05:00
test_types.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_upgrade_checkpoint.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_warnings.py Remove custom log events (#16707) 2023-02-10 14:46:26 +00:00