checkpoint_metadata.json

{
  "custom_metas": null,
  "dp": 32,
  "metas": {
    "consumed_train_samples": 51200000,
    "data_stages": [
      {
        "consumed_train_samples": 51200000,
        "name": "stable phase",
        "start_training_step": 1
      }
    ],
    "last_stage_idx": 0,
    "last_train_step": 100000
  },
  "tp": 1,
  "version": "1.4"
}