Update README.md

Update trainer