Dimensions fix

tompntn · Oct 21, 2017 · a4e454f · a4e454f
1 parent 616f5f3
commit a4e454f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/LSTM_RNN.py b/LSTM_RNN.py
@@ -91,15 +91,15 @@ def backward_pass(self, inputs, targets, first_prev_h, first_prev_c):
             'in_h': init_dict_like(self.vars['in_h']),
             'h_h': init_dict_like(self.vars['h_h']),
             'h_out': np.zeros_like(self.vars['h_out']),
-            'bias': np.zeros(OUT_DIM)
+            'bias': np.zeros(self.out_dim)
         }
 
         # Initialising Adagrad memory variables
         M_vars = {
             'in_h': init_dict_like(self.vars['in_h']),
             'h_h': init_dict_like(self.vars['h_h']),
             'h_out': np.zeros_like(self.vars['h_out']),
-            'bias': np.zeros(OUT_DIM)
+            'bias': np.zeros(self.out_dim)
         }
 
         # Backward propagation