avg_train_loss: 0.0 optimizer_step: 161304 step: 0