Use fit_generator as workaround size-mismatch bug.
Use fit_generator as workaround size-mismatch bug.

Fixes googlecolab/colabtools#709

PiperOrigin-RevId: 267172440

tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples,tensorflow/examples