cleanup

lucidrains · Mar 27, 2021 · f30f621 · f30f621
1 parent b1b00d7
commit f30f621
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/glom_pytorch/glom_pytorch.py b/glom_pytorch/glom_pytorch.py
@@ -108,7 +108,7 @@ def __init__(
         self.attention = ConsensusAttention(num_patches_side, attend_self = consensus_self, local_consensus_radius = local_consensus_radius)
 
     def forward(self, img, iters = None, levels = None, return_all = False):
-        b, h, w, _, device = *img.shape, img.device
+        b, device = img.shape[0], img.device
         iters = default(iters, self.levels * 2)   # need to have twice the number of levels of iterations in order for information to propagate up and back down. can be overridden
 
         tokens = self.image_to_tokens(img)

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'glom-pytorch',
   packages = find_packages(),
-  version = '0.0.12',
+  version = '0.0.14',
   license='MIT',
   description = 'Glom - Pytorch',
   author = 'Phil Wang',