Yes, I have specified it. It is not working
In my forward function of my model, I am taking multiple inputs. One of the inputs is an floating point, 4 of them are Variables of dimension minibatch x feature size, other is a latent vector input.
Is that could be reason for using only using 1 GPU?