Allow specifying the TrainProgram-type class for SimpleProgramSchedule.
Allow specifying the TrainProgram-type class for SimpleProgramSchedule.

This allows setting custom TrainPrograms with overrided behavior.

PiperOrigin-RevId: 320503163

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