lightning/tests/tests_pytorch
Shubhashis Roy Dipta 8663460423
Fix warning for Dataloader if num_workers = cpu count = 1 (#19224)
Co-authored-by: awaelchli <aedu.waelchli@gmail.com>
2024-01-08 09:45:02 -05:00
..
accelerators Rename PrecisionPlugin -> Precision (#18840) 2023-10-30 16:53:13 -04:00
callbacks Simplify `_get_rank()` utility function (#19220) 2024-01-02 16:24:52 +01:00
checkpointing Fix ModelCheckpoint tests from incomplete PR (#19205) 2023-12-22 08:49:09 +01:00
core Add throughput utilities to Fabric and the Trainer (#18848) 2023-10-30 17:10:29 +01:00
demos Improve handling the positional encoding in Transformer example (#18987) 2023-11-19 14:37:31 +01:00
deprecated_api TransformerEngine fallback compute dtype (#19082) 2023-12-14 03:02:09 +01:00
graveyard Rename PrecisionPlugin -> Precision (#18840) 2023-10-30 16:53:13 -04:00
helpers ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
loggers Use `step` parameter when logging metrics with NeptuneLogger (#19126) 2023-12-14 09:55:37 -05:00
loops Fix zero-grad behavior when entering the validation loop (#18710) 2023-10-09 18:07:54 -04:00
models Delay DeepSpeed config setup (#19209) 2023-12-24 17:04:04 -05:00
overrides ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
plugins Fix ModelCheckpoint tests from incomplete PR (#19205) 2023-12-22 08:49:09 +01:00
profilers Drop nvtx test (#19154) 2023-12-14 15:43:23 +01:00
serve ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
strategies Delay DeepSpeed config setup (#19209) 2023-12-24 17:04:04 -05:00
trainer Fix warning for Dataloader if num_workers = cpu count = 1 (#19224) 2024-01-08 09:45:02 -05:00
tuner Bugfix/18394 batch size finder max val batches (#18854) 2023-10-25 15:03:42 -04:00
utilities Re-enable dynamo tests that were fixed in PyTorch 2.1 (#19038) 2023-11-21 16:30:20 -05:00
__init__.py Ignore pytest cleanup warning (#19164) 2023-12-29 07:52:19 +01:00
conftest.py ci/tests: cleaning standalone script (#19141) 2023-12-13 13:27:49 -06:00
run_standalone_tasks.sh Drop nvtx test (#19154) 2023-12-14 15:43:23 +01:00
run_tpu_tests.sh ci/tests: cleaning standalone script (#19141) 2023-12-13 13:27:49 -06:00
test_cli.py Update CLI tests to no longer require 3rd party logger dependencies (#18899) 2023-10-31 09:22:17 -04:00