step train_loss 0 3.096755027770996 1 2.749246120452881 2 2.522728681564331 3 2.3631479740142822 4 2.2407546043395996