step valid_loss 0 2.919710636138916 1 2.68220853805542 2 2.524806022644043 3 2.4115779399871826 4 2.324389696121216