Fixes training command line example in README.
Fixes training command line example in README.

PiperOrigin-RevId: 266992414
Change-Id: I1a4715ad392c710b61a33e4ddc629c271e1ab95a

tensorflow/compression,tensorflow/compression