Update megaflow/utils/basic.py

megaflow/utils/basic.py

@@ -131,7 +131,7 @@ def meshgrid2d(B, Y, X, stack=False,  device='cuda', on_chans=False):
 
 def gridcloud2d(B, Y, X, norm=False, device='cuda'):
     # we want to sample for each location in the grid
-    grid_y, grid_x = meshgrid2d(B, Y, X, norm=norm, device=device)
+    grid_y, grid_x = meshgrid2d(B, Y, X, device=device)
     x = torch.reshape(grid_x, [B, -1])
     y = torch.reshape(grid_y, [B, -1])
     # these are B x N