add a handy method for ProjectionUpdater

lucidrains · Apr 21, 2021 · e40abb6 · e40abb6
1 parent 7a68c9c
commit e40abb6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/performer_pytorch/performer_pytorch.py b/performer_pytorch/performer_pytorch.py
@@ -247,6 +247,9 @@ def __init__(self, instance, feature_redraw_interval):
         self.feature_redraw_interval = feature_redraw_interval
         self.register_buffer('calls_since_last_redraw', torch.tensor(0))
 
+    def fix_projections_(self):
+        self.feature_redraw_interval = None
+
     def redraw_projections(self):
         model = self.instance
 

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'performer-pytorch',
   packages = find_packages(exclude=['examples']),
-  version = '1.0.9',
+  version = '1.0.10',
   license='MIT',
   description = 'Performer - Pytorch',
   author = 'Phil Wang',