Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -1,18 +1,18 @@
|
|
| 1 |
---
|
| 2 |
license: other
|
| 3 |
-
library_name:
|
| 4 |
pipeline_tag: robotics
|
| 5 |
---
|
| 6 |
|
| 7 |
# Cosmos Policy MetaWorld Predict2 2B
|
| 8 |
|
| 9 |
-
|
| 10 |
|
| 11 |
-
|
| 12 |
-
- `consolidated.pt`: Cosmos Policy Predict2 2B checkpoint finetuned on MetaWorld.
|
| 13 |
-
- `dataset_statistics.json`: normalization statistics
|
| 14 |
-
- `t5_embeddings.pkl`: cached text embeddings for the MetaWorld prompts.
|
| 15 |
|
| 16 |
-
The
|
| 17 |
|
| 18 |
-
The checkpoint expects
|
|
|
|
| 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.
|