Unswap train and eval streams in evaluation
Unswap train and eval streams in evaluation

PiperOrigin-RevId: 265782808

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