Update README.md

Update README.md

Model save

Training in progress, epoch 3

Training in progress, epoch 2

Training in progress, epoch 1

initial commit