checkpoints/step_20000/state.json

{"step": 20000, "loss": 6.303844451904297, "time": 1790679126.8481133}