checkpoints/step_4000/state.json

{"step": 4000, "loss": 2.552687644958496, "time": 1790662006.6080363}