[Fix] Fix flow_warp comment (open-mmlab#655)

Yshuo-Li · Dec 14, 2021 · 601d388 · 601d388
1 parent aa0a11b
commit 601d388
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmedit/models/common/flow_warp.py b/mmedit/models/common/flow_warp.py
@@ -30,7 +30,7 @@ def flow_warp(x,
     _, _, h, w = x.size()
     # create mesh grid
     grid_y, grid_x = torch.meshgrid(torch.arange(0, h), torch.arange(0, w))
-    grid = torch.stack((grid_x, grid_y), 2).type_as(x)  # (w, h, 2)
+    grid = torch.stack((grid_x, grid_y), 2).type_as(x)  # (h, w, 2)
     grid.requires_grad = False
 
     grid_flow = grid + flow