glow.py: inverse only exists on fp32+ data

NVIDIA · Apr 5, 2019 · b1469ee · b1469ee
1 parent ff9db1d
commit b1469ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/glow.py b/glow.py
@@ -88,7 +88,7 @@ def forward(self, z, reverse=False):
         if reverse:
             if not hasattr(self, 'W_inverse'):
                 # Reverse computation
-                W_inverse = W.inverse()
+                W_inverse = W.float().inverse()
                 W_inverse = Variable(W_inverse[..., None])
                 if z.type() == 'torch.cuda.HalfTensor':
                     W_inverse = W_inverse.half()