Training does not converge for ST dataset

[bbrito]

Dear @amiryanj thanks again for sharing your code. I also got issues when training with the st dataset. can you provide the parameters that you use that worked to train your model in this dataset?

Best,
Bruno