Add in predict_batch_size for train_and_eval for RetinaNet
Add in predict_batch_size for train_and_eval for RetinaNet

PiperOrigin-RevId: 256264079

tensorflow/tpu,tensorflow/tpu,tensorflow/tpu,tensorflow/tpu