Remove duplicate reshape of x1.

atong01 · Oct 9, 2024 · 5f1489d · 5f1489d
1 parent c25e191
commit 5f1489d
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/torchcfm/optimal_transport.py b/torchcfm/optimal_transport.py
@@ -81,7 +81,6 @@ def get_map(self, x0, x1):
             x0 = x0.reshape(x0.shape[0], -1)
         if x1.dim() > 2:
             x1 = x1.reshape(x1.shape[0], -1)
-        x1 = x1.reshape(x1.shape[0], -1)
         M = torch.cdist(x0, x1) ** 2
         if self.normalize_cost:
             M = M / M.max()  # should not be normalized when using minibatches