Setting specific device for Trainer

Thanks @josejames00, your solution works to me.