comment out #weights prints, to reduce noise when running tests

drowe67 · Nov 25, 2024 · 0bb2a12 · 0bb2a12
1 parent 52b412b
commit 0bb2a12
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/radae/radae_base.py b/radae/radae_base.py
@@ -185,7 +185,7 @@ def __init__(self, feature_dim, output_dim, bottleneck = 1):
         self.z_dense = nn.Linear(864, self.output_dim)
 
         nb_params = sum(p.numel() for p in self.parameters())
-        print(f"encoder: {nb_params} weights", file=sys.stderr)
+        #print(f"encoder: {nb_params} weights", file=sys.stderr)
 
         # initialize weights
         self.apply(init_weights)
@@ -251,7 +251,7 @@ def __init__(self, feature_dim, output_dim, bottleneck = 1):
         self.z_dense = nn.Linear(864, self.output_dim)
 
         nb_params = sum(p.numel() for p in self.parameters())
-        print(f"encoder: {nb_params} weights", file=sys.stderr)
+        #print(f"encoder: {nb_params} weights", file=sys.stderr)
 
         # initialize weights
         self.apply(init_weights)
@@ -326,7 +326,7 @@ def __init__(self, input_dim, output_dim):
         self.glu5 = GLU(96)
 
         nb_params = sum(p.numel() for p in self.parameters())
-        print(f"decoder: {nb_params} weights", file=sys.stderr)
+        #print(f"decoder: {nb_params} weights", file=sys.stderr)
         # initialize weights
         self.apply(init_weights)
 
@@ -393,7 +393,7 @@ def __init__(self, input_dim, output_dim):
         self.glu5 = GLU(96)
 
         nb_params = sum(p.numel() for p in self.parameters())
-        print(f"decoder: {nb_params} weights", file=sys.stderr)
+        #print(f"decoder: {nb_params} weights", file=sys.stderr)
         # initialize weights
         self.apply(init_weights)