Correct way to load pyro params to cuda


#1

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


#2

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


#3

Great! Thanks