bugfix

james · james · commit 9f7e408d5613 · 2024-08-29T22:50:51.000-07:00
diff --git a/utils/utils.py b/utils/utils.py
@@ -49,8 +49,6 @@ def forward(self, x):
         return self.apply(x)
 
     def apply(self, x):
-        nonzero = (x.abs() > self.threshold).sum()
-        print(f"Nonzero proportion: {nonzero / x.numel()}")
         return x.abs().gt(self.threshold) * x
     
     def get_threshold(self):
