srb model — Manage Model Artifacts
The srb model command packages trained checkpoints as self-contained SRB model artifacts. Artifacts carry SRB identity, the packaged checkpoint, training-dataset provenance, optional metrics, and framework restore metadata when needed.
Usage
srb model <action> [options]
Actions
| Action | Purpose |
|---|---|
create CHECKPOINT --output OUT | Package one checkpoint into a model artifact. Requires --algo; task, embodiment, and action mode can be inferred from --dataset. |
from-run RUN --output OUT | Package the latest checkpoint from a completed SRB run. The algorithm can be inferred from metadata.json. |
info PATH | Print the model manifest as JSON. |
validate PATH | Validate the artifact manifest, checkpoint, and bundled config paths. Use --strict to treat warnings as errors. |
card PATH | Write a Hugging Face README.md model card. |
push PATH | Upload or dry-run a Hugging Face model publication. |
Portable Restore
srb model from-run preserves framework restore context when the run contains it. Dreamer and TD-MPC2 artifacts can carry config.yaml; RSL-RL artifacts can carry the saved agent config from metadata.json or a nearby Hydra config. srb agent eval, srb agent collect, srb real_agent eval, and srb real_agent validate can use the artifact directory anywhere a checkpoint path is accepted.
Publication requires training-dataset provenance, either inferred from --dataset or supplied as --dataset-repo-id plus an optional --dataset-revision.
Examples
# Package the latest checkpoint from a run and infer identity from the dataset
srb model from-run ${SRB_LOGS_DIR}/excavation/rsl_rl_ppo/<run> \
--dataset /data/srb_excavation_policy \
--output /models/srb_excavation_rsl_rl_ppo
# Validate and inspect the artifact
srb model validate /models/srb_excavation_rsl_rl_ppo --strict
srb model info /models/srb_excavation_rsl_rl_ppo
# Dry-run the Hugging Face upload, including the generated README.md
srb model push /models/srb_excavation_rsl_rl_ppo --dry-run
# Reuse the packaged artifact for policy rollout
srb agent collect --env excavation --model /models/srb_excavation_rsl_rl_ppo