Result from an experiment. I have trained the model wo performing any test phase, this is to avoid the model.eval() during the training and during the saving of the model. After 30 epoch I got a train loss of <~ 2.5.
But the result is the same: both train loss and test loss are close du 6 !!!