avg_train_loss: 0.0 step: 0