Really slow

Author: lpdon

config.yml

@@ -5,5 +5,5 @@ batch_size: 8
 lr : 1.0e-4
 verbose: True
 log: train_log.txt
-save_path: 
+save_path: model.pt
 load_path: 

model.py

@@ -18,7 +18,7 @@ def __init__(self, n_classes, seq_length, batch_size):
         self.batch_size = batch_size
         self.hidden_dim = 100
         self.bidirectional = True
-        self.lstm_layers = 2
+        self.lstm_layers = 1
 
         self.conv1 = nn.Conv2d(in_channels=3, out_channels=64, kernel_size=5, padding=1, stride=1)
         self.bn1 = nn.BatchNorm2d(64)
@@ -51,7 +51,7 @@ def __init__(self, n_classes, seq_length, batch_size):
         self.bn_res3 = nn.BatchNorm2d(512)
 
         self.lstm = nn.LSTM(227328, self.hidden_dim, num_layers=self.lstm_layers, bias=True,
-                            bidirectional=self.bidirectional, dropout=0.5)
+                            bidirectional=self.bidirectional)
 
         self.fc2 = nn.Linear(self.hidden_dim * self.directions, n_classes)
         # self.fc2 = nn.Linear(self.hidden_dim, n_classes)

train.py

@@ -179,7 +179,7 @@ def train(args: Namespace, seed: int = 0, verbose: bool = False) -> Tuple[List[D
     # Load dataset and create data loaders
     dataloaders = create_dataloader(args, verbose)
 
-    seq_length = 12
+    seq_length = 15
 
     if args.load_path is not None and Path(args.load_path).is_file():
         print("Loading model weights from: " + args.load_path)