Update README.md

Update README.md

End of training

Training in progress, epoch 4

Training in progress, epoch 4

Training in progress, epoch 2

Training in progress, epoch 1

Training in progress, epoch 0

initial commit