Updates README.md to include --save_ckpt

[aksg87]

Adds --save_ckpt in the above documentation. I spent time training assuming that checkpoints would be saved by default in the pipeline, so I think adding this to the documentation would be helpful!