Skip to content

GitLab

Unverified Commit 2d7c1bb1 authored by greg2451's avatar greg2451 Committed by GitHub
Browse files

feat: add num_workers arg to DataLoader (#17751)

parent ca169dbd
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
src/transformers/trainer.py
View file @ 2d7c1bb1
...@@ -939,6 +939,7 @@ class Trainer: ...@@ -939,6 +939,7 @@ class Trainer:
batch_size=self.args.eval_batch_size, batch_size=self.args.eval_batch_size,
collate_fn=data_collator, collate_fn=data_collator,
drop_last=self.args.dataloader_drop_last, drop_last=self.args.dataloader_drop_last,
num_workers=self.args.dataloader_num_workers,
pin_memory=self.args.dataloader_pin_memory, pin_memory=self.args.dataloader_pin_memory,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment