Setting specific device for Trainer

The same issue still exist, literally dealing with this right now