diff --git a/radae/radae_base.py b/radae/radae_base.py index 95d8790..bab6236 100644 --- 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)