I’m new to Pyro, and am trying to use Pyro.MCMC to inversely quantify the uncertainty of parameters of a physical model.
I’m trying to running NUTS on GPU(cuda), I’m pretty sure I added cuda to all tensors, including pyro.sample and pyro.param, but still encountered the following error.
RuntimeError: Function AddBackward0 returned an invalid gradient at index 1 - expected type torch.cuda.FloatTensor but got torch.FloatTensor
I appreciate your help and comments on my problem. Thanks in advance!