parametrized dropout in RNN model

config.yml

@@ -58,6 +58,7 @@ RNN:
   HiddenSize: 35
   NumLayers: 4
   NumFFN: 15
+  Dropout: 0.2
   Initialization: None
 
 #Name of directory containing logs

model.py

@@ -43,6 +43,7 @@ class RNN(nn.Module):
         self.hidden_size = cfg["RNN"]["HiddenSize"]
         self.num_layers = cfg["RNN"]["NumLayers"]
         self.num_ffn = cfg["RNN"]["NumFFN"]
+        self.dropout = cfg["RNN"]["Dropout"]
         # RNN
         self.rnn = nn.Sequential(
             nn.RNN(input_size, self.hidden_size, self.num_layers, batch_first=True, nonlinearity='relu'),
@@ -62,7 +63,7 @@ class RNN(nn.Module):
             )
             self.fc.add_module(
                 f"dropout_{layer}", 
-                nn.Dropout(p=0.2)
+                nn.Dropout(p=self.dropout)
             )
         
         self.fc.add_module(