trainingarguments set device