Log the number of discretized actions only once, regarless of the train/eval batch size.
Log the number of discretized actions only once, regarless of the train/eval batch size.

PiperOrigin-RevId: 299913269

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