We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4946f55 commit 1ae744cCopy full SHA for 1ae744c
barcodebert/pretraining.py
@@ -182,6 +182,9 @@ def print_pass(*args, **kwargs):
182
eval_set = "Val"
183
184
# Dataloaders -------------------------------------------------------------
185
+ if config.cpu_workers is None:
186
+ config.cpu_workers = utils.get_num_cpu_available()
187
+
188
if config.lazy_load:
189
# streaming IterableDataset → no sampler, no shuffle
190
stream_kwargs = {
0 commit comments