Variable input and target sizes

lkulowski / LSTM_encoder_decoder

Build a LSTM encoder-decoder using PyTorch to make sequence-to-sequence prediction for time series data

MIT License

366 stars 84 forks source link

Variable input and target sizes #5

Open aatishanshuman opened 3 years ago

aatishanshuman commented 3 years ago

-Added extra argument "target_size" in init of lstm_seq2seq to allow for different sizes of input and target. -Changed the decoder input accordingly.