Allow specifying device to use when training

We should expose a way to mutate this variable to select a specific GPU to use.

changed milestone to %DAN-P5: Improve training

added P2 label

assigned to @mblanco

added Quick Win label

mentioned in merge request !299 (merged)

closed with merge request !299 (merged)