Which among the two is the right way to load the pyro parameters into cuda. Or is there any other appropriate way to go about it?
The first version should be fine. torch.autograd.Variable is deprecated and should not be used with torch>=0.4.0.
torch.autograd.Variable
torch>=0.4.0
Great! Thanks