stray print

Merge pull request #88 from lucidrains/levit
fix images
2026-03-15 01:30:15 +00:00 · 2021-04-06 13:38:52 -07:00 · 2021-04-06 13:37:46 -07:00 · 2021-04-06 13:36:26 -07:00
2 changed files with 1 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 setup(
  name = 'vit-pytorch',
  packages = find_packages(exclude=['examples']),
-  version = '0.15.0',
+  version = '0.15.1',
  license='MIT',
  description = 'Vision Transformer (ViT) - Pytorch',
  author = 'Phil Wang',
--- a/vit_pytorch/levit.py
+++ b/vit_pytorch/levit.py
@@ -81,7 +81,6 @@ class Attention(nn.Module):
    def apply_pos_bias(self, fmap):
        bias = self.pos_bias(self.pos_indices)
        bias = rearrange(bias, 'i j h -> () h i j')
-        print(bias.shape, fmap.shape)
        return fmap + bias

    def forward(self, x):
Author	SHA1	Message	Date
Phil Wang	e075460937	stray print	2021-04-06 13:38:52 -07:00
Phil Wang	5e23e48e4d	Merge pull request #88 from lucidrains/levit fix images	2021-04-06 13:37:46 -07:00
Phil Wang	0f31ca79e3	Merge pull request #87 from lucidrains/levit levit without pos emb	2021-04-06 13:36:26 -07:00