step valid_loss 0 2.890321969985962 1 2.669679880142212 2 2.5183584690093994 3 2.4061686992645264 4 2.3184266090393066