Fix header image

Update README.md

Update README.md

Model save

Training in progress, step 610

Training in progress, step 600

Training in progress, step 300

initial commit