Update README.md

Update README.md

Update README.md

Update README.md

Update README.md

Update README.md

Update README.md

Training complete

Training complete

End of training

Model save

Training in progress, epoch 0

Training in progress, epoch 1

Training in progress, epoch 1

Training in progress, epoch 0

initial commit