How to put autoguide parameters on GPU

I m running simple bayesian model using self defined model function, and autoguide function such as AutoIAFNormal, it always says 'Expected object of backend CPU but got backend CUDA for argument #4 ‘mat1’

I put model in net and set = autoguide

I just simply don’t know how to put parameters from guide to GPU

there was a bug that has been fixed in dev, so you can either use the dev version of pyro or torch.set_default_tensor_type('torch.cuda.FloatTensor') (or some other tensor type)