Remove print

vballoli · Mar 3, 2021 · 0e06842 · 0e06842
1 parent c57fc7b
commit 0e06842
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/nfnets/agc.py b/nfnets/agc.py
@@ -104,7 +104,6 @@ def zero_grad(self, set_to_none: bool = False):
         for group in self.agc_params:
             for p in group['params']:
                 if p.grad is not None:
-                    print("Zeroing")
                     if set_to_none:
                         p.grad = None
                     else:

diff --git a/setup.py b/setup.py
@@ -8,7 +8,7 @@
 setup(
   name = 'nfnets-pytorch',
   packages = find_packages(),
-  version = '0.1.0',
+  version = '0.1.1',
   license='MIT',
   description = 'NFNets, PyTorch',
   long_description=long_description,