pruned unrelated comments

MiscellaneousStuff · May 5, 2022 · 0b9301b · 0b9301b
1 parent 48b6623
commit 0b9301b
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/model.py b/model.py
@@ -110,16 +110,12 @@ def __init__(self, n_cnn_layers, n_rnn_layers, rnn_dim, n_class, n_feats, stride
  )
 
  def forward(self, x):
- # print("PRE MODEL INPUT SHAPE:", x.shape)
  x = self.cnn(x)
  x = self.rescnn_layers(x)
- # print("POST MODEL INPUT SHAPE:", x.shape)
  sizes = x.size()
  x = x.view(sizes[0], sizes[1] * sizes[2], sizes[3]) # (batch, feature, time)
- # print("VIEWED MODEL INPUT SHAPE:", x.shape)
  x = x.transpose(1, 2) # (batch, time, feature)
  x = self.fully_connected(x)
  x = self.birnn_layers(x)
  x = self.classifier(x)
- # print("POST-SEQ SHAPE:", x.shape)
  return x
diff --git a/train.py b/train.py