[FIX] print statement in baseline model

NINAnor · Feb 20, 2024 · 70cb5da · 70cb5da
1 parent aafd3db
commit 70cb5da
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/Models/baseline.py b/Models/baseline.py
@@ -24,7 +24,6 @@ def __init__(self):
     def forward(self,x):
         (num_samples,seq_len,mel_bins) = x.shape
         x = x.view(-1,1,seq_len,mel_bins)
-        print(x.shape)
         x = self.encoder(x)
         x = nn.MaxPool2d(2)(x)