allow fixed params in hmc

t-vi · Nov 24, 2017 · bd48929 · bd48929
1 parent 4d19cc5
commit bd48929
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/candlegp/training/hmc.py b/candlegp/training/hmc.py
@@ -50,7 +50,7 @@ def hmc_sample(model, num_samples, epsilon, lmin=1, lmax=2, thin=1, burn=0):
 
     def logprob_grads():
         logprob = -model.objective()
-        grads = torch.autograd.grad(logprob, model.parameters())
+        grads = torch.autograd.grad(logprob, [p for p in model.parameters() if p.requires_grad])
         return logprob, grads
 
     def thinning(thin_iterations, epsilon, lmin, lmax):