Save gin config after first epoch in RL Trainer (so it's easy to get back to experiment parameters).
Save gin config after first epoch in RL Trainer (so it's easy to get back to experiment parameters).

PiperOrigin-RevId: 313083441

google/trax,google/trax