step train_loss 0 3.0726168155670166 1 2.7409346103668213 2 2.5224294662475586 3 2.364570140838623 4 2.2422571182250977