Merge pull request #3 from facebookresearch/fix_integer_multiplication

Fix Integer multiplication done before float conversion
facebookresearch · Aug 6, 2019 · 1603926 · 1603926
2 parents 6a7ccb1 + 61ed582
commit 1603926
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/videoalignment/temporal_match_kernel.py b/videoalignment/temporal_match_kernel.py
@@ -89,7 +89,7 @@ def __init__(
             raise NotImplementedError
 
         self.a = nn.Parameter(torch.from_numpy(np_a).float())  # (T, m)
-        self.ms = (2 * np.pi * torch.arange(0, self.m)).float()
+        self.ms = 2 * np.pi * torch.arange(0, self.m).float()
         self.Ts = torch.tensor(self.T, dtype=torch.float32, requires_grad=False)
 
     def single_fv(self, ts, xs):