LSTM_LN

LSTM_LN copied to clipboard

batch first mode causes some error

6

`if __name__ == '__main__': model = LSTM(50, 100, 2,batch_first=True) x = Variable(Tensor(32,50, 50)) h = model.init_hidden(32) print(model(x, h))` and it will return : `Traceback (most recent call last):' 'File "lstm.py",...

GuanlinLee

see https://github.com/exe1023/LSTM_LN/issues/1#issuecomment-586294848

NiziL

I tried replacing a regular pytorch lstm with this implementation and I get this error. I am not quite sure what I am supposed to put as the 'hiddens' argment...

martinper