fix meshgrid issue

lucidrains · Dec 3, 2021 · 64ca548 · 64ca548
1 parent 6be0c37
commit 64ca548
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'siren-pytorch',
   packages = find_packages(),
-  version = '0.1.5',
+  version = '0.1.6',
   license='MIT',
   description = 'Implicit Neural Representations with Periodic Activation Functions',
   author = 'Phil Wang',

diff --git a/siren_pytorch/siren_pytorch.py b/siren_pytorch/siren_pytorch.py
@@ -133,8 +133,8 @@ def __init__(self, net, image_width, image_height, latent_dim = None):
                 num_layers = net.num_layers
             )
 
-        tensors = [torch.linspace(-1, 1, steps = image_width), torch.linspace(-1, 1, steps = image_height)]
-        mgrid = torch.stack(torch.meshgrid(*tensors), dim=-1)
+        tensors = [torch.linspace(-1, 1, steps = image_height), torch.linspace(-1, 1, steps = image_width)]
+        mgrid = torch.stack(torch.meshgrid(*tensors, indexing = 'ij'), dim=-1)
         mgrid = rearrange(mgrid, 'h w c -> (h w) c')
         self.register_buffer('grid', mgrid)