Allow selecting GPU in train_net