diff --git a/src/ptbench/data/datamodule.py b/src/ptbench/data/datamodule.py index 3e699347607db040631c6ae6540412750ddede86..5c94091877ff735e74942795203aa82c8931b61e 100644 --- a/src/ptbench/data/datamodule.py +++ b/src/ptbench/data/datamodule.py @@ -429,7 +429,7 @@ class CachingDataModule(lightning.LightningDataModule): the system. Set to >= 1 to enable that many multiprocessing instances for data loading. - It sets the parameter ``num_workers`` (from Dataloaders) to match the + It sets the parameter ``num_workers`` (from DataLoaders) to match the expected pytorch representation. For macOS machines, it also sets the ``multiprocessing_context`` to use ``spawn`` instead of the default.