train_state.json

{
  "phase": 3,
  "step": 22,
  "best_kl": 0.16501659924785297
}