first_layer_1/model_4000/training_state.json

{
    "global_step": 4000,
    "update_step": 4000,
    "tokens_seen": 2555953152,
    "tokens_seen_before": 2555314176,
    "update_time": 0.9765069484710693,
    "wandb_id": "krzb2185"
}