steps 5 - epoch 0 - loss 0.11006805300712585 - lr 0.0