checkpoints/020000/training_state/training_step.json

{
    "step": 20000,
    "num_processes": 3,
    "batch_size": 32
}