We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fd92ab9 commit 7a9e37aCopy full SHA for 7a9e37a
src/lightning/data/processing/functions.py
@@ -73,7 +73,7 @@ def _get_input_dir(inputs: Sequence[Any]) -> Optional[str]:
73
def _get_default_num_workers() -> int:
74
if torch.cuda.is_available():
75
return torch.cuda.device_count()
76
- return os.cpu_count()
+ return os.cpu_count() or 1
77
78
79
class LambdaDataTransformRecipe(DataTransformRecipe):
0 commit comments