checkpoints/step_2000/state.json

{"step": 2000, "loss": 3.0559544563293457, "time": 1790660489.3182886}