Setting specific device for Trainer

Beginners
12

Meanwhile, this can do a trick:

trainer = Trainer(
    model = model,
    .
    .
    device=0 # id of the device to use (int) -> for GPU(s): 0, 1 or for CPU: -1
)