| { |
| "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" |
| } |
|
|
