Correct way to load pyro params to cuda

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?


or

The first version should be fine. torch.autograd.Variable is deprecated and should not be used with torch>=0.4.0.

Great! Thanks