{
  "checkpoint_schema": {
    "config": "minimal model-loading config (model and vocab)",
    "epoch": "completed training epoch",
    "format_version": "integer",
    "model": "complete CellWorld state_dict",
    "scale": "small | base | large | huge",
    "step": "completed optimizer step"
  },
  "format_version": 1,
  "intended_visibility": "private",
  "models": {
    "base": {
      "checkpoint": "base/model.pt",
      "checkpoint_bytes": 109565803,
      "checkpoint_sha256": "567f90bed46919c857e6674e3cac1712feedc8ee976febbab0c57c26705b65ec",
      "config": "base/config.yaml",
      "config_sha256": "177dd96fe1cd7966572f143ef36c107372bc3a849beb7f1458960ccf7d84f514",
      "epoch": 110,
      "state_tensor_elements": 27372864,
      "step": 10000
    },
    "huge": {
      "checkpoint": "huge/model.pt",
      "checkpoint_bytes": 680899563,
      "checkpoint_sha256": "6001593e462a1f487494fbcfd78513216c54649e8767bb6ee1bc7825b56cf397",
      "config": "huge/config.yaml",
      "config_sha256": "90293b34a55bc84297a94eedf9ea859ebb523515bed64a700f2fc0dd529cabcd",
      "epoch": 110,
      "state_tensor_elements": 170191488,
      "step": 10000
    },
    "large": {
      "checkpoint": "large/model.pt",
      "checkpoint_bytes": 248771115,
      "checkpoint_sha256": "77f1eb38eb8ddaa6e1e8fa987653b187840a848220a49a748dae16b7225c6ef7",
      "config": "large/config.yaml",
      "config_sha256": "c7e3cec75ca1244fe64d19262279cda34c75d0cd18e4dcb31f61088a8066f515",
      "epoch": 110,
      "state_tensor_elements": 62166784,
      "step": 10000
    },
    "small": {
      "checkpoint": "small/model.pt",
      "checkpoint_bytes": 30146155,
      "checkpoint_sha256": "6fa0e149a024d3fe11ba8386a5d91af64fedc15fd2bceade5035603839ea3929",
      "config": "small/config.yaml",
      "config_sha256": "52d9389933d823a37057b569d97847600be05d4e2683faafb6753853bcf6b9d9",
      "epoch": 110,
      "state_tensor_elements": 7521696,
      "step": 10000
    }
  },
  "repository": "Haiping-UoM/CellWorld"
}