Question: how to use TorchConnector and specify gpu devices

[zuxfoucault]

In this qgan example (https://qiskit-community.github.io/qiskit-machine-learning/tutorials/04_torch_qgan.html)
how do I specify gpu in TorchConnector?
or just simply with

device = 'GPU' if 'GPU' in AerSimulator().available_devices() else 'CPU'
sim = AerSimulator(method='statevector', device=device)
sampler = AerSampler(
      options={'backend_options':{'method': sim.options.method, 
                                    'device' : sim.options.device, 
                                    'cuStateVec_enable' : True}}
    )

[edoaltamura]

Indeed, that's the way to do it if you want to simulate the training part of the algorithm. @OkuyanBoga @oscar-wallis do you have tips for selecting the device from TorchConnector as an alternative?