lightning/tests/tests_pytorch/utilities
Carlos Mocholí 8c4184c105
Integrate with `lightning_utilities.core.enums` (#14558)
2022-09-07 15:14:14 +02:00
..
__init__.py
test_all_gather_grad.py
test_argparse.py Remove the legacy `get_deprecated_arg_names` (#14415) 2022-08-29 14:53:57 +02:00
test_auto_restart.py fix imports of collections.abc for py3.10 (#14345) 2022-08-23 11:52:58 -04:00
test_data.py Allowed setting attributes on `DataLoader` and `BatchSampler` when instantiated inside `*_dataloader` hooks (#14212) 2022-08-17 11:42:54 -04:00
test_deepspeed_collate_checkpoint.py
test_deepspeed_model_summary.py Organize model summary utilities (#13893) 2022-07-28 19:23:29 +02:00
test_device_parser.py Add support for DDP fork (#13405) 2022-07-22 16:05:35 +00:00
test_distributed.py Update gather_all_tensors to handle tensors of different sizes (#12630) 2022-06-29 17:03:00 +00:00
test_dtype_device_mixin.py Mark the lite `DeviceDtypeModuleMixin` as protected (#14548) 2022-09-06 14:17:15 +02:00
test_enums.py Integrate with `lightning_utilities.core.enums` (#14558) 2022-09-07 15:14:14 +02:00
test_fetching.py Remove mps config for test (#14379) 2022-08-26 02:47:37 -04:00
test_finite_checks.py
test_grads.py Fix dtype inference during gradient norm computation (#14051) 2022-08-08 11:35:06 +00:00
test_imports.py Integrate `lightning_utilities.core.imports` (#14475) 2022-09-06 12:56:20 +00:00
test_logger.py save checkpoints and profiler output to the first logger (#14325) 2022-08-26 17:23:54 +00:00
test_memory.py
test_migration.py
test_model_helpers.py
test_model_summary.py Organize model summary utilities (#13893) 2022-07-28 19:23:29 +02:00
test_optimizer.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_parameter_tying.py
test_parsing.py Fix `LightningDataModule` hparams parsing (#12806) 2022-08-25 18:57:48 +00:00
test_rank_zero.py Integrate `lightning_utilities.core.rank_zero` (#14556) 2022-09-07 09:21:48 +00:00
test_remote_filesystem.py
test_seed.py update rng state save/load test to also run on cuda gpu (#14396) 2022-08-31 16:36:35 +00:00
test_torchdistx.py Integrate `lightning_utilities.core.imports` (#14475) 2022-09-06 12:56:20 +00:00
test_types.py
test_upgrade_checkpoint.py
test_warnings.py Integrate `lightning_utilities.core.rank_zero` (#14556) 2022-09-07 09:21:48 +00:00