diff --git a/model.py b/model.py
index ba54a29..eccac3e 100644
--- 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
\ No newline at end of file
diff --git a/train.py b/train.py
new file mode 100644
index 0000000..e69de29