Introduce a variable logging hook which will log the values of all variables during training via a session run  hook.
Introduce a variable logging hook which will log the values of all variables during training via a session run  hook.

PiperOrigin-RevId: 276103839

google-research/tensor2robot