Silence warnings in PyTorch 1.10

HolyWu · Oct 26, 2021 · ea2a597 · ea2a597
1 parent 886f695
commit ea2a597
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vsbasicvsrpp/flow_warp.py b/vsbasicvsrpp/flow_warp.py
@@ -29,7 +29,7 @@ def flow_warp(x,
                          f'flow ({flow.size()[1:3]}) are not the same.')
     _, _, h, w = x.size()
     # create mesh grid
-    grid_y, grid_x = torch.meshgrid(torch.arange(0, h), torch.arange(0, w))
+    grid_y, grid_x = torch.meshgrid(torch.arange(0, h), torch.arange(0, w), indexing='ij')
     grid = torch.stack((grid_x, grid_y), 2).type_as(x)  # (w, h, 2)
     grid.requires_grad = False