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