Update conv_lstm.py

Fixed input format to be (batch, seq_len,Chans,H,W)
SojeongAn · Sep 10, 2017 · 890d8e6 · 890d8e6
1 parent b74e76a
commit 890d8e6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/conv_lstm.py b/conv_lstm.py
@@ -84,8 +84,8 @@ def forward(self, input, hidden_state):
 
         """
 
-        #current_input = input.transpose(0, 1)#now is seq_len,B,C,H,W
-        current_input=input
+        current_input = input.transpose(0, 1)#now is seq_len,B,C,H,W
+        #current_input=input
         next_hidden=[]#hidden states(h and c)
         seq_len=current_input.size(0)