Fixing #11

Apparently a version change bug!
adavoudi · May 3, 2020 · 2a15e90 · 2a15e90
1 parent 3c4e3f9
commit 2a15e90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spdnet/spd.py b/spdnet/spd.py
@@ -293,7 +293,7 @@ def backward(ctx, grad_output):
 
                 max_mask = s > epsilon
                 s_max_diag = s.clone(); s_max_diag[~max_mask] = epsilon; s_max_diag = s_max_diag.diag()
-                Q = max_mask.diag().float()
+                Q = max_mask.float().diag()
 
                 dLdV = 2*(g.mm(u.mm(s_max_diag)))
                 dLdS = eye * (Q.mm(u.t().mm(g.mm(u))))