am3338

I managed to overcome this issue by decreasing the learning rate to **2e-5** in the training script. However, the results I obtain are still somewhat worse than what is reported...