lightning/tests/tests_pytorch/utilities
Leng Yue 6f4524a25c
Support kwargs input for `LayerSummary` (#17709)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2023-05-30 00:17:50 -04:00
..
migration ruff: autofix PT (#17541) 2023-05-04 11:50:39 -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: enable & fixing RET (#17540) 2023-05-05 09:34:40 +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 avoid unnecessary workers with sequential `CombinedLoader ` (#17639) 2023-05-30 04:02:50 +00:00
test_compile.py Add dynamo RunIf skip condition (#17404) 2023-04-19 01:09:42 +02:00
test_data.py Replace IPU with external implementation (#17075) 2023-04-27 16:09:51 +00:00
test_deepspeed_collate_checkpoint.py [TPU] Fix PjRT tests (#17408) 2023-04-19 16:39:00 +02:00
test_deepspeed_model_summary.py Introduce new precision layout in PL (#16783) 2023-02-17 17:58:14 +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 ruff: PT some more fixes (#17569) 2023-05-05 08:25:15 +02:00
test_imports.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_memory.py Remove unused code (#17001) 2023-03-09 00:07:48 +00:00
test_model_helpers.py Check for mixed new and old style imports (#17548) 2023-05-05 13:16:19 +02:00
test_model_summary.py Support kwargs input for `LayerSummary` (#17709) 2023-05-30 00:17:50 -04:00
test_parameter_tying.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_parsing.py Make the `is_picklable` function more robust (#17270) 2023-04-06 22:20:52 +00: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_signature_utils.py Support inspecting the signature of decorated hooks (#17507) 2023-04-28 09:42:45 +00:00
test_torchdistx.py ruff: PT some more fixes (#17569) 2023-05-05 08:25:15 +02:00
test_types.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_upgrade_checkpoint.py Add map location option to checkpoint upgrade utility (#17527) 2023-04-29 13:55:41 +00:00
test_warnings.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00