| .. |
|
__init__.py
|
…
|
|
|
test_apply_func.py
|
ruff: replace isort with ruff +TPU (#17684)
|
2023-09-26 11:54:55 -04:00 |
|
test_cloud_io.py
|
precommit: ruff-format (#19434)
|
2024-02-15 13:39:17 -05:00 |
|
test_consolidate_checkpoint.py
|
Utility to consolidate sharded checkpoints (#19213)
|
2024-01-23 17:15:22 -05:00 |
|
test_data.py
|
Refactor BoringFabric in tests (#19364)
|
2024-01-30 23:32:45 +01:00 |
|
test_device_dtype_mixin.py
|
ruff: replace isort with ruff +TPU (#17684)
|
2023-09-26 11:54:55 -04:00 |
|
test_device_parser.py
|
ruff: replace isort with ruff +TPU (#17684)
|
2023-09-26 11:54:55 -04:00 |
|
test_distributed.py
|
Update tests for PyTorch 2.2.1 (#19521)
|
2024-02-23 13:11:34 -05:00 |
|
test_imports.py
|
ci: testing with py3.11 (#17822)
|
2023-08-28 15:15:12 +00:00 |
|
test_init.py
|
Drop support for PyTorch 1.12 (#19300)
|
2024-01-26 11:44:24 -05:00 |
|
test_load.py
|
Utility to consolidate sharded checkpoints (#19213)
|
2024-01-23 17:15:22 -05:00 |
|
test_logger.py
|
Flatten dataclass hyperparameters for logging (#18906)
|
2023-11-03 19:30:19 -04:00 |
|
test_optimizer.py
|
ruff: replace isort with ruff +TPU (#17684)
|
2023-09-26 11:54:55 -04:00 |
|
test_rank_zero.py
|
PyTest random order for Fabric tests (#19040)
|
2023-11-22 16:41:49 -05:00 |
|
test_registry.py
|
External callback registry through entry points for Fabric (#17756)
|
2023-06-06 11:53:19 +00:00 |
|
test_runif.py
|
Compose RunIf utilities (#17520)
|
2023-05-05 01:21:58 +02:00 |
|
test_seed.py
|
Request `torch.cuda` RNG states only if CUDA is available (#19234)
|
2024-01-10 16:16:29 -05:00 |
|
test_spike.py
|
Update tests for PyTorch 2.2.1 (#19521)
|
2024-02-23 13:11:34 -05:00 |
|
test_throughput.py
|
Fallback to `ACCELERATOR_TYPE` for TPU flops (#19314)
|
2024-01-24 16:21:56 +01:00 |
|
test_warnings.py
|
precommit: ruff-format (#19434)
|
2024-02-15 13:39:17 -05:00 |
a41528c2a6