Minor cleanup in trainer_model_free.py
Minor cleanup in trainer_model_free.py

PiperOrigin-RevId: 231845132

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