avg_train_loss: 0.0 optimizer_step: 166520 step: 0