fixing simple reward in sv2p.
fixing simple reward in sv2p.

PiperOrigin-RevId: 213731557

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