Remove logging of raw tensors

MadryLab · Jan 1, 2024 · 73d16d5 · 73d16d5
1 parent 2554f5f
commit 73d16d5
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/trak/traker.py b/trak/traker.py
@@ -475,10 +475,8 @@ def finalize_features(
 
             g = ch.as_tensor(self.saver.current_store["grads"], device=self.device)
             xtx = self.score_computer.get_xtx(g)
-            self.logger.debug(f"XTX is {xtx}")
 
             features = self.score_computer.get_x_xtx_inv(g, xtx)
-            self.logger.debug(f"Features are {features}")
             self.saver.current_store["features"][:] = features.to(self.dtype).cpu()
             if del_grads:
                 self.saver.del_grads(model_id)