Mayank Malkoti

change hidden_size in train.py to 54. You see this is a simple matrix multiplication issue. The row of second matrix should match the column of the first matrix.