Skip to content

Commit 1da6965

Browse files
authored
Chronos-2: set default dataloader_num_workers=0 in Trainer (#409)
*Issue #, if available:* *Description of changes:* 0 is a better default than 1. By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.
1 parent 4eea8d0 commit 1da6965

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/chronos/chronos2/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -265,7 +265,7 @@ def fit(
265265
report_to="none",
266266
max_steps=num_steps,
267267
gradient_accumulation_steps=1,
268-
dataloader_num_workers=1,
268+
dataloader_num_workers=0,
269269
tf32=has_sm80 and not use_cpu,
270270
bf16=has_sm80 and not use_cpu,
271271
save_only_model=True,

0 commit comments

Comments
 (0)