Change number of workers.

[danielsoy]

Hey!

Got this warning: UserWarning: This DataLoader will create 8 worker processes in total

I need to reduce the number of workers from 8, to 4.

How do i change this variable in run_training.py?

Thanks!

[Runinho]

You can use the --workers argument:

python run_training.py --model_dir models --head_layer 2 --cuda 1 --workers 4

Or hardcode it here:

workers=4)

[danielsoy]

Thanks!