training_state.json

{
  "global_step": 16000,
  "best_loss": Infinity,
  "stage": "stage2"
}