• Chady Kamar's avatar
    Add num workers cli arg (#7322) · 17099ebd
    Chady Kamar authored
    * Add dataloader_num_workers to TrainingArguments
    
    This argument is meant to be used to set the
    number of workers for the PyTorch DataLoader.
    
    * Pass num_workers argument on DataLoader init
    17099ebd
trainer.py 66.5 KB