Some minor improvements in `train.py`
Some minor improvements in `train.py`

* Some typing / naming improvements
* Short circuit to have empty eval inputs if not running eval during train (`eval_interval_steps == 0 `)

PiperOrigin-RevId: 488981684

google/paxml,google/paxml