should never require gradients

lucidrains · lucidrains · commit 37620e4ae34a · 2023-03-06T17:50:01.000-08:00
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'vector_quantize_pytorch',
   packages = find_packages(),
-  version = '1.1.0',
+  version = '1.1.1',
   license='MIT',
   description = 'Vector Quantization - Pytorch',
   long_description_content_type = 'text/markdown',
diff --git a/vector_quantize_pytorch/random_projection_quantizer.py b/vector_quantize_pytorch/random_projection_quantizer.py
@@ -34,6 +34,7 @@ def __init__(
             **kwargs
         )
 
+    @torch.no_grad()
     def forward(self, x):
 
         x = einsum('b n d, h d e -> b n h e', x, self.rand_projs)

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,7 @@ def __init__(`
`34`	`34`	`**kwargs`
`35`	`35`	`)`
`36`	`36`
	`37`	`+ @torch.no_grad()`
`37`	`38`	`def forward(self, x):`
`38`	`39`
`39`	`40`	`x = einsum('b n d, h d e -> b n h e', x, self.rand_projs)`