train_state.json

{
  "phase": 1,
  "step": 9,
  "best_kl": 0.17905689477920533
}