avg_train_loss: 0.0 optimizer_step: 178480 step: 0