Haongchen commited on
Commit
3b27f9a
·
verified ·
1 Parent(s): 1bb78ad

Document exact runtime config paths

Browse files
Files changed (1) hide show
  1. README.md +0 -147
README.md CHANGED
@@ -1,147 +0,0 @@
1
- ---
2
- library_name: memoryvla
3
- tags:
4
- - robotics
5
- - vision-language-action
6
- - memory
7
- - action-diffusion
8
- - piper
9
- pipeline_tag: robotics
10
- license: apache-2.0
11
- ---
12
-
13
- # MemoryVLA
14
-
15
- MemoryVLA checkpoint for real-robot action prediction. This repository is
16
- organized so that additional real-world task checkpoints can be added under
17
- `checkpoints/<task-name>/` without replacing the current model.
18
-
19
- ## Latest checkpoint
20
-
21
- The current default checkpoint is:
22
-
23
- ```text
24
- checkpoints/piper-color-sorting/frozen-ema-step-20000.pt
25
- ```
26
-
27
- It is the EMA action-diffusion checkpoint from the Frozen+EMA training run.
28
- The matching action normalization statistics are stored at:
29
-
30
- ```text
31
- configs/piper-color-sorting/dataset_statistics.json
32
- ```
33
-
34
- The checkpoint is intended to be loaded with the MemoryVLA codebase and
35
- `use_ema=True`. The model uses the `custom_finetuning` normalization key.
36
-
37
- ## Training configuration
38
-
39
- - Task: Piper color sorting
40
- - Data: 21 valid real-world episodes, 14,300 frames
41
- - Observation: front camera only in this dataset; no wrist-camera stream was
42
- available in the training data
43
- - Base VLM: `prism-dinosiglip-224px+7b`
44
- - Vision backbone: frozen
45
- - LLM backbone: frozen
46
- - Last LLM layer: frozen
47
- - Trainable components: action diffusion model and MemoryVLA trainable
48
- modules
49
- - Action dimension: 7 (`x, y, z, roll, pitch, yaw, gripper`)
50
- - Action model: `DiT-L`
51
- - Future action window: 15
52
- - Memory length: 256
53
- - Retrieval layers: 2
54
- - Fusion: `gate`
55
- - Consolidation: `tome`
56
- - Per-device batch size: 2
57
- - Global batch size: 32
58
- - Optimizer learning rate: `2e-5`
59
- - Scheduler: linear warmup + cosine decay
60
- - Warmup ratio: 0.03
61
- - EMA: enabled, decay `0.999`
62
- - Mixed precision: enabled
63
- - Training strategy: FSDP full shard
64
- - Maximum training steps: 20,000
65
- - Image augmentation: disabled
66
- - Random seed: 42
67
-
68
- ## Open-loop evaluation
69
-
70
- On the same 21 training episodes, using frame-by-frame memory-aware inference:
71
-
72
- - Normalized overall action RMSE: `0.2124`
73
- - Gripper accuracy: `98.64%`
74
- - Memory reset: at the first frame of every episode
75
- - Cognitive and perception memory banks: capped at 256 entries
76
-
77
- These are training-set open-loop results and should not be interpreted as
78
- unseen-task generalization.
79
-
80
- ## Loading outline
81
-
82
- The exact loader depends on the MemoryVLA code revision. The essential
83
- settings for this checkpoint are:
84
-
85
- ```python
86
- model = load_vla(
87
- model_id_or_path="checkpoints/piper-color-sorting/frozen-ema-step-20000.pt",
88
- load_for_training=False,
89
- action_dim=7,
90
- future_action_window_size=15,
91
- action_model_type="DiT-L",
92
- mem_length=256,
93
- retrieval_layers=2,
94
- use_timestep_pe=True,
95
- fusion_type="gate",
96
- consolidate_type="tome",
97
- update_fused=False,
98
- use_ema=True,
99
- )
100
- ```
101
-
102
- For real deployment, reset the episode memory before the first observation of
103
- each task and pass the matching `dataset_statistics.json` when unnormalizing
104
- actions. The gripper action is sign-encoded (`-1` / `+1`) by the current
105
- pipeline.
106
-
107
- ## Repository layout
108
-
109
- ```text
110
- checkpoints/
111
- piper-color-sorting/
112
- frozen-ema-step-20000.pt
113
- configs/
114
- piper-color-sorting/
115
- dataset_statistics.json
116
- ```
117
-
118
- Future tasks should use a separate directory, for example:
119
-
120
- ```text
121
- checkpoints/
122
- piper-color-sorting/
123
- drawer-opening/
124
- peg-insertion/
125
- configs/
126
- piper-color-sorting/
127
- drawer-opening/
128
- peg-insertion/
129
- ```
130
-
131
- Keep the checkpoint, action statistics, task name, camera convention, and
132
- training configuration together for every task.
133
-
134
- ## Code and reproducibility
135
-
136
- The Hugging Face repository stores model artifacts and deployment metadata.
137
- A separate GitHub repository is recommended for the MemoryVLA model code,
138
- real-robot wrapper, preprocessing, and evaluation scripts. It is not required
139
- to download the weights, but it makes future task training, deployment, and
140
- exact code-version tracking much safer. Record the Git commit or release tag
141
- used for each checkpoint in the corresponding task directory or Model Card.
142
-
143
- ## Intended use
144
-
145
- Research and development for real-robot manipulation. Validate workspace
146
- limits, action scaling, emergency stop behavior, camera calibration, and
147
- gripper sign conventions before sending actions to hardware.