File size: 833 Bytes
c2b5ddd f665938 c2b5ddd f665938 c2b5ddd f665938 c2b5ddd f665938 c2b5ddd f665938 c2b5ddd f665938 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | ---
license: other
---
# VGGT-Omega 1B 256 OXE/MimicGen/RoboCasa365 Step 62000
This repo contains the portable model checkpoint `0062000.pt` and the DeepSpeed ZeRO-2 checkpoint shards needed for optimizer-state resume.
Important: because this run used DeepSpeed ZeRO-2, `0062000.pt` intentionally stores `optimizer: null`. Optimizer state lives in `checkpoints/0062000/`.
To resume with optimizer/scheduler state in this codebase, keep the sibling layout:
```text
0062000.pt
checkpoints/0062000/
bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt
...
bf16_zero_pp_rank_7_mp_rank_00_optim_states.pt
mp_rank_00_model_states.pt
```
Then pass `--ckpt /path/to/0062000.pt` with the `checkpoints/0062000/` directory present; `train_robot.py` resolves the sibling ZeRO folder and calls DeepSpeed `load_checkpoint(tag="0062000")`.
|