train_state.json

{
  "step": 98000,
  "loss": 4.706041845286987
}