Setting specific device for Trainer

[Harsh786]

Yes, you’re correct @yaanhaan, but if you pass this argument into the Trainer, It’ll use it for TrainingArguments that’ll set the given device for you run.

It worked for me BTW