Correct typo affecting multi-problem training.
Correct typo affecting multi-problem training.

PiperOrigin-RevId: 228432054

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