Update README.md

End of training

Model save

End of training

Training in progress, step 480

Training in progress, step 360

Training in progress, step 240

Training in progress, step 120

Training in progress, step 180

Training in progress, step 120

Training in progress, step 60

initial commit