Fix a NaN bug in SimPLe, caused by incomplete batches in the training input stream
Fix a NaN bug in SimPLe, caused by incomplete batches in the training input stream

PiperOrigin-RevId: 266265965

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor