fix grads

locuslab · Feb 4, 2020 · cf8251e · cf8251e
1 parent 0eed02f
commit cf8251e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lml.py b/lml.py
@@ -164,7 +164,8 @@ def backward(ctx, grad_output):
             dx = torch.zeros_like(x)
             if single:
                 dx = dx.squeeze()
-            return dx
+            grads = tuple([dx] + [None]*5)
+            return grads
 
         Hinv = 1./(1./y + 1./(1.-y))
         dnu = bdot(Hinv, grad_output)/Hinv.sum(dim=1)