Error when decoder has more than 1 layer.

[pajola]

texar-pytorch/examples/vae_text/config_lstm_yahoo.py

Line 62 in 0ba18bf

"num_layers": 1

The output is the follwoing:
RuntimeError: Input batch size 128 doesn't match hidden[0] batch size 256

The issue is due to the "initial_state=lstm_states" when the decoder is forwarded.