AlexWortega commited on
Commit
8a154bd
·
verified ·
1 Parent(s): 23c5675

Upload folder using huggingface_hub

Browse files
tinyvla2/pretrain_C_incontext/README.md ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # TinyVLA-2 — C-incontext (in-context demo conditioning, NEGATIVE RESULT)
2
+
3
+ K=3 (obs,action) exemplar pairs from the target robot fed as conditioning tokens
4
+ at test time, no gradient updates. Trained 20k steps from C-qwen-morph.
5
+
6
+ ## Result: dead channel
7
+ Real vs garbage/noise support examples give indistinguishable results (324.7mm vs
8
+ 322.5mm) — the model learned to ignore demo content. Likely cause: during training,
9
+ support pairs were sampled from the SAME source dataset as the query, redundant
10
+ with the already-available embodiment identity signal (text+numeric channels), so
11
+ gradient found no reason to depend on demo content.
12
+
13
+ **Do not use this checkpoint** — C-diverse remains the reference (150mm on
14
+ held-out jaco, 295mm zero-shot LeKiwi). Full context: tinyvla2/RESULTS.md.
tinyvla2/pretrain_C_incontext/config.json ADDED
@@ -0,0 +1,98 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "type": "tinyvla",
3
+ "n_obs_steps": 1,
4
+ "input_features": {
5
+ "observation.images.cam0": {
6
+ "type": "VISUAL",
7
+ "shape": [
8
+ 3,
9
+ 256,
10
+ 256
11
+ ]
12
+ },
13
+ "observation.images.cam1": {
14
+ "type": "VISUAL",
15
+ "shape": [
16
+ 3,
17
+ 256,
18
+ 256
19
+ ]
20
+ },
21
+ "observation.state": {
22
+ "type": "STATE",
23
+ "shape": [
24
+ 16
25
+ ]
26
+ }
27
+ },
28
+ "output_features": {
29
+ "action": {
30
+ "type": "ACTION",
31
+ "shape": [
32
+ 8
33
+ ]
34
+ }
35
+ },
36
+ "device": "cuda",
37
+ "use_amp": false,
38
+ "use_peft": false,
39
+ "push_to_hub": true,
40
+ "repo_id": null,
41
+ "private": null,
42
+ "tags": null,
43
+ "license": null,
44
+ "pretrained_path": null,
45
+ "pretrained_revision": null,
46
+ "chunk_size": 50,
47
+ "n_action_steps": 50,
48
+ "normalization_mapping": {
49
+ "VISUAL": "IDENTITY",
50
+ "STATE": "MEAN_STD",
51
+ "ACTION": "MEAN_STD"
52
+ },
53
+ "max_state_dim": 16,
54
+ "max_action_dim": 8,
55
+ "num_embodiments": 16,
56
+ "max_cameras": 3,
57
+ "conditioning": "morph_qwen",
58
+ "morph_tokens": 2,
59
+ "morph_to_slow": false,
60
+ "num_morph_readout": 4,
61
+ "morph_text_max_len": 32,
62
+ "use_demo_conditioning": true,
63
+ "n_support": 3,
64
+ "action_space": "canonical",
65
+ "image_size": 256,
66
+ "freeze_vision_encoder": true,
67
+ "lm_model_name": "Qwen/Qwen3.5-0.8B",
68
+ "lm_num_layers": 12,
69
+ "num_readout_tokens": 8,
70
+ "freeze_lm": false,
71
+ "tokenizer_max_length": 48,
72
+ "pad_language_to": "longest",
73
+ "expert_dim": 512,
74
+ "expert_layers": 12,
75
+ "expert_heads": 8,
76
+ "use_semantic_latent": true,
77
+ "use_spatial_tokens": true,
78
+ "spatial_vocab": 1024,
79
+ "spatial_loss_weight": 0.0,
80
+ "num_denoise_steps": 10,
81
+ "flow_beta_alpha": 1.5,
82
+ "flow_beta_beta": 1.0,
83
+ "min_period": 0.004,
84
+ "max_period": 4.0,
85
+ "staleness_prob": 0.0,
86
+ "staleness_max_s": 2.0,
87
+ "optimizer_lr": 0.0001,
88
+ "optimizer_betas": [
89
+ 0.9,
90
+ 0.95
91
+ ],
92
+ "optimizer_eps": 1e-08,
93
+ "optimizer_weight_decay": 1e-10,
94
+ "optimizer_grad_clip_norm": 10.0,
95
+ "scheduler_warmup_steps": 1000,
96
+ "scheduler_decay_steps": 40000,
97
+ "scheduler_decay_lr": 2.5e-06
98
+ }
tinyvla2/pretrain_C_incontext/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4f01fdbff1f33230680fb3da16050c79fc43317f372895f95796d0f6dd172ec
3
+ size 2578929032