fix train code

Update README.md

Update README.md

Update README.md

Add files via upload

Update README.md

Update README.md

Update README.md

Update README.md

Update train_sr.sh

Add files via upload

Delete __main__

update

Add files via upload

Update train_sr.sh