Updated helper function output format; add a test for the CPU trainer checkpoints
Updated helper function output format; add a test for the CPU trainer checkpoints

PiperOrigin-RevId: 405490597

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo