mmurray-ms commited on
Commit
e5e4a54
·
verified ·
1 Parent(s): 535dc0c

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -1,18 +1,18 @@
1
  ---
2
  license: other
3
- library_name: pytorch
4
  pipeline_tag: robotics
5
  ---
6
 
7
  # Cosmos Policy MetaWorld Predict2 2B
8
 
9
- Artifacts for the FlowDAgger OSS Cosmos/MetaWorld assembly example.
10
 
11
- This bundle contains:
12
- - `consolidated.pt`: Cosmos Policy Predict2 2B checkpoint finetuned on MetaWorld.
13
- - `dataset_statistics.json`: normalization statistics used by the policy wrapper.
14
- - `t5_embeddings.pkl`: cached text embeddings for the MetaWorld prompts.
15
 
16
- The released FlowDAgger recipe targets MetaWorld `assembly-v3`. In the July 2026 reproduction, the frozen base policy reached about 52% success over 25 evaluation episodes and FlowDAgger full-mode steering reached about 84-88%.
17
 
18
- The checkpoint expects the MetaWorld gripper proprio convention `[g, -g]`; use `--gripper_qpos_mode opposed` in the OSS launcher.
 
1
  ---
2
  license: other
3
+ library_name: cosmos-policy
4
  pipeline_tag: robotics
5
  ---
6
 
7
  # Cosmos Policy MetaWorld Predict2 2B
8
 
9
+ FlowDAgger OSS release artifact for the Cosmos 2B MetaWorld backend.
10
 
11
+ Files:
12
+ - `consolidated.pt`: Cosmos Policy Predict2 2B checkpoint finetuned on MetaWorld data for the release recipe.
13
+ - `dataset_statistics.json`: action/proprio normalization statistics.
14
+ - `t5_embeddings.pkl`: cached text embeddings for the MetaWorld prompts used by the example.
15
 
16
+ The release recipe targets MetaWorld `assembly-v3` with prompt `Pick up a nut and place it onto a peg`. This checkpoint intentionally leaves headroom: the frozen base policy is about 52% SR over 25 eval episodes in the reference run, and FlowDAgger full-mode steering reaches about 84-88%.
17
 
18
+ The checkpoint expects MetaWorld gripper proprio encoded as `[g, g]`; use `--gripper_qpos_mode duplicated` with the FlowDAgger Cosmos runner.