{
  "step": 62000,
  "single_file_checkpoint": "0062000.pt",
  "zero2_checkpoint_dir": "checkpoints/0062000",
  "optimizer_resume_note": "The single .pt has optimizer=None under DeepSpeed ZeRO-2. Use --ckpt 0062000.pt with sibling checkpoints/0062000/ present so train_robot.py can call DeepSpeed load_checkpoint(tag=0062000).",
  "required_zero2_files": [
    "bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt",
    "bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt",
    "bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt",
    "bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt",
    "bf16_zero_pp_rank_4_mp_rank_00_optim_states.pt",
    "bf16_zero_pp_rank_5_mp_rank_00_optim_states.pt",
    "bf16_zero_pp_rank_6_mp_rank_00_optim_states.pt",
    "bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt",
    "mp_rank_00_model_states.pt"
  ]
}