Hugging Face's logo

Daankrol
/
snapflow-multi-obj

Model card Files
xet
 Community
Daankrol commited on
Commit
ebc903d
·
verified ·
1 Parent(s): d1afc15

Upload folder using huggingface_hub

Browse files
Files changed (29) hide show
  1. .gitattributes +2 -0
  2. phase2_snapflow/last/exports/openvino/manifest.json +218 -0
  3. phase2_snapflow/last/exports/openvino/pi05.bin +3 -0
  4. phase2_snapflow/last/exports/openvino/pi05.xml +0 -0
  5. phase2_snapflow/last/exports/openvino/tokenizer.bin +3 -0
  6. phase2_snapflow/last/exports/openvino/tokenizer.xml +790 -0
  7. phase2_snapflow/snapflow-epoch001/exports/openvino/manifest.json +218 -0
  8. phase2_snapflow/snapflow-epoch001/exports/openvino/pi05.bin +3 -0
  9. phase2_snapflow/snapflow-epoch001/exports/openvino/pi05.xml +0 -0
  10. phase2_snapflow/snapflow-epoch001/exports/openvino/tokenizer.bin +3 -0
  11. phase2_snapflow/snapflow-epoch001/exports/openvino/tokenizer.xml +790 -0
  12. phase2_snapflow/wandb/debug-internal.log +0 -0
  13. phase2_snapflow/wandb/debug.log +21 -0
  14. phase2_snapflow/wandb/run-20260806_235610-358tcehl/files/config.yaml +375 -0
  15. phase2_snapflow/wandb/run-20260806_235610-358tcehl/files/output.log +0 -0
  16. phase2_snapflow/wandb/run-20260806_235610-358tcehl/files/requirements.txt +209 -0
  17. phase2_snapflow/wandb/run-20260806_235610-358tcehl/files/wandb-metadata.json +71 -0
  18. phase2_snapflow/wandb/run-20260806_235610-358tcehl/files/wandb-summary.json +1 -0
  19. phase2_snapflow/wandb/run-20260806_235610-358tcehl/logs/debug-core.log +27 -0
  20. phase2_snapflow/wandb/run-20260806_235610-358tcehl/logs/debug-internal.log +0 -0
  21. phase2_snapflow/wandb/run-20260806_235610-358tcehl/logs/debug.log +23 -0
  22. phase2_snapflow/wandb/run-20260806_235610-358tcehl/run-358tcehl.wandb +3 -0
  23. phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/files/output.log +0 -0
  24. phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/files/requirements.txt +158 -0
  25. phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/files/wandb-metadata.json +72 -0
  26. phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/logs/debug-core.log +7 -0
  27. phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/logs/debug-internal.log +0 -0
  28. phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/logs/debug.log +21 -0
  29. phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/run-7i8cqirl.wandb +3 -0
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ phase2_snapflow/wandb/run-20260806_235610-358tcehl/run-358tcehl.wandb filter=lfs diff=lfs merge=lfs -text
37
+ phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/run-7i8cqirl.wandb filter=lfs diff=lfs merge=lfs -text
phase2_snapflow/last/exports/openvino/manifest.json ADDED
@@ -0,0 +1,218 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "policy": {
3
+ "name": "pi05",
4
+ "source": {
5
+ "class_path": "physicalai.policies.pi05.policy.Pi05"
6
+ }
7
+ },
8
+ "model": {
9
+ "runner": {
10
+ "class_path": "physicalai.inference.runners.single_pass.SinglePass"
11
+ },
12
+ "artifacts": {
13
+ "openvino": "pi05.xml"
14
+ },
15
+ "preprocessors": [
16
+ {
17
+ "type": "normalize",
18
+ "stats": {
19
+ "state": {
20
+ "mean": [
21
+ 0.4902085707550287,
22
+ -15.791447722340527,
23
+ 18.025905360949306,
24
+ 55.00042447703383,
25
+ -49.21564963228982,
26
+ 9.598709224247822
27
+ ],
28
+ "std": [
29
+ 28.56218428477719,
30
+ 44.62872421028026,
31
+ 43.38945126960788,
32
+ 19.454495761579995,
33
+ 8.781436382129517,
34
+ 8.932825748525975
35
+ ],
36
+ "min": [
37
+ -71.17327117919922,
38
+ -99.66344451904297,
39
+ -56.68328094482422,
40
+ -6.712270259857178,
41
+ -85.72921752929688,
42
+ 0.26385223865509033
43
+ ],
44
+ "max": [
45
+ 80.10877990722656,
46
+ 73.74842071533203,
47
+ 98.82192993164062,
48
+ 92.56092071533203,
49
+ -3.136435031890869,
50
+ 42.084434509277344
51
+ ],
52
+ "q01": [
53
+ -39.45999045040392,
54
+ -98.81794802740971,
55
+ -23.48391958469011,
56
+ 25.50333895589101,
57
+ -61.34767717791836,
58
+ 0.6040729837518146
59
+ ],
60
+ "q99": [
61
+ 34.18319318983752,
62
+ 39.30228886401892,
63
+ 98.56118793779315,
64
+ 82.42731621305099,
65
+ -36.652074374338476,
66
+ 26.018496216201445
67
+ ],
68
+ "type": "STATE",
69
+ "name": "state",
70
+ "shape": [
71
+ 6
72
+ ]
73
+ }
74
+ },
75
+ "mode": "quantiles"
76
+ },
77
+ {
78
+ "type": "pi05",
79
+ "image_resolution": [
80
+ 224,
81
+ 224
82
+ ],
83
+ "empty_cameras": 0
84
+ },
85
+ {
86
+ "type": "ov_tokenizer",
87
+ "artifact": "tokenizer.xml"
88
+ }
89
+ ],
90
+ "postprocessors": [
91
+ {
92
+ "type": "denormalize",
93
+ "stats": {
94
+ "action": {
95
+ "mean": [
96
+ 0.47757554785419337,
97
+ -18.034652611064505,
98
+ 15.59805635303217,
99
+ 54.54463619743119,
100
+ -49.21284246567293,
101
+ 7.302610387584375
102
+ ],
103
+ "std": [
104
+ 29.091115680540447,
105
+ 44.16383911983223,
106
+ 44.81775979436524,
107
+ 19.76430732442855,
108
+ 8.799572987673075,
109
+ 10.41276650889744
110
+ ],
111
+ "min": [
112
+ -75.1649169921875,
113
+ -100.0,
114
+ -61.050724029541016,
115
+ -6.849315166473389,
116
+ -86.33736419677734,
117
+ 0.0
118
+ ],
119
+ "max": [
120
+ 81.91695404052734,
121
+ 72.79933166503906,
122
+ 100.0,
123
+ 93.32191467285156,
124
+ -2.732527494430542,
125
+ 42.88956069946289
126
+ ],
127
+ "q01": [
128
+ -41.29065982641405,
129
+ -100.0,
130
+ -27.74406375139188,
131
+ 24.550995379247542,
132
+ -61.591176330648565,
133
+ -1.000000013351432e-10
134
+ ],
135
+ "q99": [
136
+ 34.99315845909429,
137
+ 38.49093164574106,
138
+ 99.99756689137342,
139
+ 82.60630455329574,
140
+ -36.44491567201352,
141
+ 26.6719993779817
142
+ ],
143
+ "type": "ACTION",
144
+ "name": "action",
145
+ "shape": [
146
+ 6
147
+ ]
148
+ }
149
+ },
150
+ "mode": "quantiles"
151
+ }
152
+ ],
153
+ "input_features": [
154
+ {
155
+ "class_path": "physicalai.inference.data.features.InferenceFeature",
156
+ "init_args": {
157
+ "ftype": "STATE",
158
+ "shape": [
159
+ 6
160
+ ],
161
+ "name": "state",
162
+ "dtype": "float32"
163
+ }
164
+ },
165
+ {
166
+ "class_path": "physicalai.inference.data.features.InferenceFeature",
167
+ "init_args": {
168
+ "ftype": "VISUAL",
169
+ "shape": [
170
+ 3,
171
+ 480,
172
+ 640
173
+ ],
174
+ "name": "images.gripper",
175
+ "dtype": "float32"
176
+ }
177
+ },
178
+ {
179
+ "class_path": "physicalai.inference.data.features.InferenceFeature",
180
+ "init_args": {
181
+ "ftype": "VISUAL",
182
+ "shape": [
183
+ 3,
184
+ 480,
185
+ 640
186
+ ],
187
+ "name": "images.overhead",
188
+ "dtype": "float32"
189
+ }
190
+ },
191
+ {
192
+ "class_path": "physicalai.inference.data.features.InferenceFeature",
193
+ "init_args": {
194
+ "ftype": "LANGUAGE",
195
+ "shape": [],
196
+ "name": "task",
197
+ "dtype": "string"
198
+ }
199
+ }
200
+ ],
201
+ "output_features": [
202
+ {
203
+ "class_path": "physicalai.inference.data.features.InferenceFeature",
204
+ "init_args": {
205
+ "ftype": "ACTION",
206
+ "shape": [
207
+ 50,
208
+ 6
209
+ ],
210
+ "name": "action",
211
+ "dtype": "float32"
212
+ }
213
+ }
214
+ ]
215
+ },
216
+ "format": "policy_package",
217
+ "version": "1.0"
218
+ }
phase2_snapflow/last/exports/openvino/pi05.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3418fbcb1ff70c217d9003ddb2659927a9b7111658f2072daef0d565b211d803
3
+ size 6252282268
phase2_snapflow/last/exports/openvino/pi05.xml ADDED
The diff for this file is too large to render. See raw diff
 
phase2_snapflow/last/exports/openvino/tokenizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:817945e2aced98317970c4c588906f3b702ea2ace8300540e897274881a8a6a8
3
+ size 17932006
phase2_snapflow/last/exports/openvino/tokenizer.xml ADDED
@@ -0,0 +1,790 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ <?xml version="1.0"?>
2
+ <net name="tokenizer" version="11">
3
+ <layers>
4
+ <layer id="0" name="Parameter_40982" type="Parameter" version="opset1">
5
+ <data shape="?" element_type="string" />
6
+ <output>
7
+ <port id="0" precision="STRING" names="Parameter_40982">
8
+ <dim>-1</dim>
9
+ </port>
10
+ </output>
11
+ </layer>
12
+ <layer id="1" name="Constant_41101" type="Const" version="opset1">
13
+ <data element_type="i32" shape="" offset="0" size="4" />
14
+ <output>
15
+ <port id="0" precision="I32" />
16
+ </output>
17
+ </layer>
18
+ <layer id="2" name="Constant_41102" type="Const" version="opset1">
19
+ <data element_type="i32" shape="" offset="4" size="4" />
20
+ <output>
21
+ <port id="0" precision="I32" />
22
+ </output>
23
+ </layer>
24
+ <layer id="3" name="Constant_41103" type="Const" version="opset1">
25
+ <data element_type="i32" shape="1" offset="8" size="4" />
26
+ <output>
27
+ <port id="0" precision="I32">
28
+ <dim>1</dim>
29
+ </port>
30
+ </output>
31
+ </layer>
32
+ <layer id="4" name="Constant_40988" type="Const" version="opset1">
33
+ <data element_type="i64" shape="" offset="12" size="8" />
34
+ <output>
35
+ <port id="0" precision="I64" />
36
+ </output>
37
+ </layer>
38
+ <layer id="5" name="StringTensorUnpack_40983" type="StringTensorUnpack" version="opset15">
39
+ <input>
40
+ <port id="0" precision="STRING">
41
+ <dim>-1</dim>
42
+ </port>
43
+ </input>
44
+ <output>
45
+ <port id="1" precision="I32">
46
+ <dim>-1</dim>
47
+ </port>
48
+ <port id="2" precision="I32">
49
+ <dim>-1</dim>
50
+ </port>
51
+ <port id="3" precision="U8">
52
+ <dim>-1</dim>
53
+ </port>
54
+ </output>
55
+ </layer>
56
+ <layer id="6" name="ShapeOf_40984" type="ShapeOf" version="opset3">
57
+ <data output_type="i64" />
58
+ <input>
59
+ <port id="0" precision="I32">
60
+ <dim>-1</dim>
61
+ </port>
62
+ </input>
63
+ <output>
64
+ <port id="1" precision="I64">
65
+ <dim>1</dim>
66
+ </port>
67
+ </output>
68
+ </layer>
69
+ <layer id="7" name="Constant_40985" type="Const" version="opset1">
70
+ <data element_type="i64" shape="" offset="12" size="8" />
71
+ <output>
72
+ <port id="0" precision="I64" />
73
+ </output>
74
+ </layer>
75
+ <layer id="8" name="Constant_40986" type="Const" version="opset1">
76
+ <data element_type="i64" shape="" offset="12" size="8" />
77
+ <output>
78
+ <port id="0" precision="I64" />
79
+ </output>
80
+ </layer>
81
+ <layer id="9" name="Gather_40987" type="Gather" version="opset8">
82
+ <data batch_dims="0" />
83
+ <input>
84
+ <port id="0" precision="I64">
85
+ <dim>1</dim>
86
+ </port>
87
+ <port id="1" precision="I64" />
88
+ <port id="2" precision="I64" />
89
+ </input>
90
+ <output>
91
+ <port id="3" precision="I64" />
92
+ </output>
93
+ </layer>
94
+ <layer id="10" name="Constant_40989" type="Const" version="opset1">
95
+ <data element_type="i64" shape="" offset="20" size="8" />
96
+ <output>
97
+ <port id="0" precision="I64" />
98
+ </output>
99
+ </layer>
100
+ <layer id="11" name="Range_40990" type="Range" version="opset4">
101
+ <data output_type="i32" />
102
+ <input>
103
+ <port id="0" precision="I64" />
104
+ <port id="1" precision="I64" />
105
+ <port id="2" precision="I64" />
106
+ </input>
107
+ <output>
108
+ <port id="3" precision="I32">
109
+ <dim>-1</dim>
110
+ </port>
111
+ </output>
112
+ </layer>
113
+ <layer id="12" name="Constant_40991" type="Const" version="opset1">
114
+ <data element_type="i64" shape="" offset="20" size="8" />
115
+ <output>
116
+ <port id="0" precision="I64" />
117
+ </output>
118
+ </layer>
119
+ <layer id="13" name="Constant_40992" type="Const" version="opset1">
120
+ <data element_type="i64" shape="" offset="20" size="8" />
121
+ <output>
122
+ <port id="0" precision="I64" />
123
+ </output>
124
+ </layer>
125
+ <layer id="14" name="Add_40993" type="Add" version="opset1">
126
+ <data auto_broadcast="numpy" />
127
+ <input>
128
+ <port id="0" precision="I64" />
129
+ <port id="1" precision="I64" />
130
+ </input>
131
+ <output>
132
+ <port id="2" precision="I64" />
133
+ </output>
134
+ </layer>
135
+ <layer id="15" name="Constant_40994" type="Const" version="opset1">
136
+ <data element_type="i64" shape="" offset="20" size="8" />
137
+ <output>
138
+ <port id="0" precision="I64" />
139
+ </output>
140
+ </layer>
141
+ <layer id="16" name="Range_40995" type="Range" version="opset4">
142
+ <data output_type="i32" />
143
+ <input>
144
+ <port id="0" precision="I64" />
145
+ <port id="1" precision="I64" />
146
+ <port id="2" precision="I64" />
147
+ </input>
148
+ <output>
149
+ <port id="3" precision="I32">
150
+ <dim>-1</dim>
151
+ </port>
152
+ </output>
153
+ </layer>
154
+ <layer id="17" name="Constant_41059" type="Const" version="opset1">
155
+ <data element_type="u8" shape="4365" offset="28" size="4365" />
156
+ <output>
157
+ <port id="0" precision="U8">
158
+ <dim>4365</dim>
159
+ </port>
160
+ </output>
161
+ </layer>
162
+ <layer id="18" name="SpecialTokensSplit_41060" type="SpecialTokensSplit" version="extension">
163
+ <input>
164
+ <port id="0" precision="I32">
165
+ <dim>-1</dim>
166
+ </port>
167
+ <port id="1" precision="I32">
168
+ <dim>-1</dim>
169
+ </port>
170
+ <port id="2" precision="I32">
171
+ <dim>-1</dim>
172
+ </port>
173
+ <port id="3" precision="I32">
174
+ <dim>-1</dim>
175
+ </port>
176
+ <port id="4" precision="U8">
177
+ <dim>-1</dim>
178
+ </port>
179
+ <port id="5" precision="U8">
180
+ <dim>4365</dim>
181
+ </port>
182
+ </input>
183
+ <output>
184
+ <port id="6" precision="I32">
185
+ <dim>-1</dim>
186
+ </port>
187
+ <port id="7" precision="I32">
188
+ <dim>-1</dim>
189
+ </port>
190
+ <port id="8" precision="I32">
191
+ <dim>-1</dim>
192
+ </port>
193
+ <port id="9" precision="I32">
194
+ <dim>-1</dim>
195
+ </port>
196
+ <port id="10" precision="U8">
197
+ <dim>-1</dim>
198
+ </port>
199
+ <port id="11" precision="BOOL">
200
+ <dim>-1</dim>
201
+ </port>
202
+ </output>
203
+ </layer>
204
+ <layer id="19" name="Constant_41062" type="Const" version="opset1">
205
+ <data element_type="u8" shape="1" offset="4393" size="1" />
206
+ <output>
207
+ <port id="0" precision="U8">
208
+ <dim>1</dim>
209
+ </port>
210
+ </output>
211
+ </layer>
212
+ <layer id="20" name="Constant_41064" type="Const" version="opset1">
213
+ <data element_type="u8" shape="3" offset="4394" size="3" />
214
+ <output>
215
+ <port id="0" precision="U8">
216
+ <dim>3</dim>
217
+ </port>
218
+ </output>
219
+ </layer>
220
+ <layer id="21" name="RegexNormalization_41065" type="RegexNormalization" version="extension">
221
+ <data global_replace="true" />
222
+ <input>
223
+ <port id="0" precision="I32">
224
+ <dim>-1</dim>
225
+ </port>
226
+ <port id="1" precision="I32">
227
+ <dim>-1</dim>
228
+ </port>
229
+ <port id="2" precision="U8">
230
+ <dim>-1</dim>
231
+ </port>
232
+ <port id="3" precision="BOOL">
233
+ <dim>-1</dim>
234
+ </port>
235
+ <port id="4" precision="U8">
236
+ <dim>1</dim>
237
+ </port>
238
+ <port id="5" precision="U8">
239
+ <dim>3</dim>
240
+ </port>
241
+ </input>
242
+ <output>
243
+ <port id="6" precision="I32">
244
+ <dim>-1</dim>
245
+ </port>
246
+ <port id="7" precision="I32">
247
+ <dim>-1</dim>
248
+ </port>
249
+ <port id="8" precision="U8">
250
+ <dim>-1</dim>
251
+ </port>
252
+ <port id="9" precision="BOOL">
253
+ <dim>-1</dim>
254
+ </port>
255
+ </output>
256
+ </layer>
257
+ <layer id="22" name="Constant_41067" type="Const" version="opset1">
258
+ <data element_type="u8" shape="1" offset="4393" size="1" />
259
+ <output>
260
+ <port id="0" precision="U8">
261
+ <dim>1</dim>
262
+ </port>
263
+ </output>
264
+ </layer>
265
+ <layer id="23" name="RegexSplit_41068" type="RegexSplit" version="extension">
266
+ <data behaviour="mergedwithprevious" invert="false" max_splits="-1" />
267
+ <input>
268
+ <port id="0" precision="I32">
269
+ <dim>-1</dim>
270
+ </port>
271
+ <port id="1" precision="I32">
272
+ <dim>-1</dim>
273
+ </port>
274
+ <port id="2" precision="I32">
275
+ <dim>-1</dim>
276
+ </port>
277
+ <port id="3" precision="I32">
278
+ <dim>-1</dim>
279
+ </port>
280
+ <port id="4" precision="U8">
281
+ <dim>-1</dim>
282
+ </port>
283
+ <port id="5" precision="BOOL">
284
+ <dim>-1</dim>
285
+ </port>
286
+ <port id="6" precision="U8">
287
+ <dim>1</dim>
288
+ </port>
289
+ </input>
290
+ <output>
291
+ <port id="7" precision="I32">
292
+ <dim>-1</dim>
293
+ </port>
294
+ <port id="8" precision="I32">
295
+ <dim>-1</dim>
296
+ </port>
297
+ <port id="9" precision="I32">
298
+ <dim>-1</dim>
299
+ </port>
300
+ <port id="10" precision="I32">
301
+ <dim>-1</dim>
302
+ </port>
303
+ <port id="11" precision="U8">
304
+ <dim>-1</dim>
305
+ </port>
306
+ <port id="12" precision="BOOL">
307
+ <dim>-1</dim>
308
+ </port>
309
+ </output>
310
+ </layer>
311
+ <layer id="24" name="Constant_41070" type="Const" version="opset1">
312
+ <data element_type="i32" shape="257153" offset="4397" size="1028612" />
313
+ <output>
314
+ <port id="0" precision="I32">
315
+ <dim>257153</dim>
316
+ </port>
317
+ </output>
318
+ </layer>
319
+ <layer id="25" name="Constant_41072" type="Const" version="opset1">
320
+ <data element_type="i32" shape="257153" offset="1033009" size="1028612" />
321
+ <output>
322
+ <port id="0" precision="I32">
323
+ <dim>257153</dim>
324
+ </port>
325
+ </output>
326
+ </layer>
327
+ <layer id="26" name="Constant_41074" type="Const" version="opset1">
328
+ <data element_type="u8" shape="1942155" offset="2061621" size="1942155" />
329
+ <output>
330
+ <port id="0" precision="U8">
331
+ <dim>1942155</dim>
332
+ </port>
333
+ </output>
334
+ </layer>
335
+ <layer id="27" name="Constant_41082" type="Const" version="opset1">
336
+ <data element_type="i32" shape="580604" offset="4003776" size="2322416" />
337
+ <output>
338
+ <port id="0" precision="I32">
339
+ <dim>580604</dim>
340
+ </port>
341
+ </output>
342
+ </layer>
343
+ <layer id="28" name="Constant_41084" type="Const" version="opset1">
344
+ <data element_type="i32" shape="580604" offset="6326192" size="2322416" />
345
+ <output>
346
+ <port id="0" precision="I32">
347
+ <dim>580604</dim>
348
+ </port>
349
+ </output>
350
+ </layer>
351
+ <layer id="29" name="Constant_41086" type="Const" version="opset1">
352
+ <data element_type="u8" shape="2709312" offset="8648608" size="2709312" />
353
+ <output>
354
+ <port id="0" precision="U8">
355
+ <dim>2709312</dim>
356
+ </port>
357
+ </output>
358
+ </layer>
359
+ <layer id="30" name="Constant_41088" type="Const" version="opset1">
360
+ <data element_type="i32" shape="580604" offset="11357920" size="2322416" />
361
+ <output>
362
+ <port id="0" precision="I32">
363
+ <dim>580604</dim>
364
+ </port>
365
+ </output>
366
+ </layer>
367
+ <layer id="31" name="Constant_41090" type="Const" version="opset1">
368
+ <data element_type="i32" shape="580604" offset="13680336" size="2322416" />
369
+ <output>
370
+ <port id="0" precision="I32">
371
+ <dim>580604</dim>
372
+ </port>
373
+ </output>
374
+ </layer>
375
+ <layer id="32" name="Constant_41092" type="Const" version="opset1">
376
+ <data element_type="u8" shape="1923319" offset="16002752" size="1923319" />
377
+ <output>
378
+ <port id="0" precision="U8">
379
+ <dim>1923319</dim>
380
+ </port>
381
+ </output>
382
+ </layer>
383
+ <layer id="33" name="Constant_41076" type="Const" version="opset1">
384
+ <data element_type="i32" shape="217" offset="17926071" size="868" />
385
+ <output>
386
+ <port id="0" precision="I32">
387
+ <dim>217</dim>
388
+ </port>
389
+ </output>
390
+ </layer>
391
+ <layer id="34" name="Constant_41078" type="Const" version="opset1">
392
+ <data element_type="i32" shape="217" offset="17926939" size="868" />
393
+ <output>
394
+ <port id="0" precision="I32">
395
+ <dim>217</dim>
396
+ </port>
397
+ </output>
398
+ </layer>
399
+ <layer id="35" name="Constant_41080" type="Const" version="opset1">
400
+ <data element_type="u8" shape="3305" offset="17927807" size="3305" />
401
+ <output>
402
+ <port id="0" precision="U8">
403
+ <dim>3305</dim>
404
+ </port>
405
+ </output>
406
+ </layer>
407
+ <layer id="36" name="Constant_41093" type="Const" version="opset1">
408
+ <data element_type="i32" shape="217" offset="17931112" size="868" />
409
+ <output>
410
+ <port id="0" precision="I32">
411
+ <dim>217</dim>
412
+ </port>
413
+ </output>
414
+ </layer>
415
+ <layer id="37" name="BPETokenizer_41094" type="BPETokenizer" version="extension">
416
+ <data unk_token="&lt;unk>" fuse_unk="true" suffix_indicator="" end_suffix="" byte_fallback="true" cache_capacity="51430" />
417
+ <input>
418
+ <port id="0" precision="I32">
419
+ <dim>-1</dim>
420
+ </port>
421
+ <port id="1" precision="I32">
422
+ <dim>-1</dim>
423
+ </port>
424
+ <port id="2" precision="I32">
425
+ <dim>-1</dim>
426
+ </port>
427
+ <port id="3" precision="I32">
428
+ <dim>-1</dim>
429
+ </port>
430
+ <port id="4" precision="U8">
431
+ <dim>-1</dim>
432
+ </port>
433
+ <port id="5" precision="I32">
434
+ <dim>257153</dim>
435
+ </port>
436
+ <port id="6" precision="I32">
437
+ <dim>257153</dim>
438
+ </port>
439
+ <port id="7" precision="U8">
440
+ <dim>1942155</dim>
441
+ </port>
442
+ <port id="8" precision="I32">
443
+ <dim>580604</dim>
444
+ </port>
445
+ <port id="9" precision="I32">
446
+ <dim>580604</dim>
447
+ </port>
448
+ <port id="10" precision="U8">
449
+ <dim>2709312</dim>
450
+ </port>
451
+ <port id="11" precision="I32">
452
+ <dim>580604</dim>
453
+ </port>
454
+ <port id="12" precision="I32">
455
+ <dim>580604</dim>
456
+ </port>
457
+ <port id="13" precision="U8">
458
+ <dim>1923319</dim>
459
+ </port>
460
+ <port id="14" precision="I32">
461
+ <dim>217</dim>
462
+ </port>
463
+ <port id="15" precision="I32">
464
+ <dim>217</dim>
465
+ </port>
466
+ <port id="16" precision="U8">
467
+ <dim>3305</dim>
468
+ </port>
469
+ <port id="17" precision="I32">
470
+ <dim>217</dim>
471
+ </port>
472
+ </input>
473
+ <output>
474
+ <port id="18" precision="I32">
475
+ <dim>-1</dim>
476
+ </port>
477
+ <port id="19" precision="I32">
478
+ <dim>-1</dim>
479
+ </port>
480
+ <port id="20" precision="I32">
481
+ <dim>-1</dim>
482
+ </port>
483
+ </output>
484
+ </layer>
485
+ <layer id="38" name="Constant_41095" type="Const" version="opset1">
486
+ <data element_type="i32" shape="" offset="17931980" size="4" />
487
+ <output>
488
+ <port id="0" precision="I32" />
489
+ </output>
490
+ </layer>
491
+ <layer id="39" name="Constant_41097" type="Const" version="opset1">
492
+ <data element_type="u8" shape="5" offset="17931984" size="5" />
493
+ <output>
494
+ <port id="0" precision="U8">
495
+ <dim>5</dim>
496
+ </port>
497
+ </output>
498
+ </layer>
499
+ <layer id="40" name="Constant_41099" type="Const" version="opset1">
500
+ <data element_type="u8" shape="13" offset="17931989" size="13" />
501
+ <output>
502
+ <port id="0" precision="U8">
503
+ <dim>13</dim>
504
+ </port>
505
+ </output>
506
+ </layer>
507
+ <layer id="41" name="Truncate_41100" type="Truncate" version="extension">
508
+ <data m_num_inputs="1" />
509
+ <input>
510
+ <port id="0" precision="I32">
511
+ <dim>-1</dim>
512
+ </port>
513
+ <port id="1" precision="I32">
514
+ <dim>-1</dim>
515
+ </port>
516
+ <port id="2" precision="I32">
517
+ <dim>-1</dim>
518
+ </port>
519
+ <port id="3" precision="I32" />
520
+ <port id="4" precision="U8">
521
+ <dim>5</dim>
522
+ </port>
523
+ <port id="5" precision="U8">
524
+ <dim>13</dim>
525
+ </port>
526
+ </input>
527
+ <output>
528
+ <port id="6" precision="I32">
529
+ <dim>-1</dim>
530
+ </port>
531
+ <port id="7" precision="I32">
532
+ <dim>-1</dim>
533
+ </port>
534
+ <port id="8" precision="I32">
535
+ <dim>-1</dim>
536
+ </port>
537
+ </output>
538
+ </layer>
539
+ <layer id="42" name="Constant_41104" type="Const" version="opset1">
540
+ <data element_type="i32" shape="2" offset="12" size="8" />
541
+ <output>
542
+ <port id="0" precision="I32">
543
+ <dim>2</dim>
544
+ </port>
545
+ </output>
546
+ </layer>
547
+ <layer id="43" name="CombineSegments_41105" type="CombineSegments" version="extension">
548
+ <input>
549
+ <port id="0" precision="I32" />
550
+ <port id="1" precision="I32" />
551
+ <port id="2" precision="I32">
552
+ <dim>1</dim>
553
+ </port>
554
+ <port id="3" precision="I32">
555
+ <dim>-1</dim>
556
+ </port>
557
+ <port id="4" precision="I32">
558
+ <dim>-1</dim>
559
+ </port>
560
+ <port id="5" precision="I32">
561
+ <dim>-1</dim>
562
+ </port>
563
+ <port id="6" precision="I32">
564
+ <dim>2</dim>
565
+ </port>
566
+ </input>
567
+ <output>
568
+ <port id="7" precision="I32">
569
+ <dim>-1</dim>
570
+ </port>
571
+ <port id="8" precision="I32">
572
+ <dim>-1</dim>
573
+ </port>
574
+ <port id="9" precision="I32">
575
+ <dim>-1</dim>
576
+ </port>
577
+ <port id="10" precision="I32">
578
+ <dim>-1</dim>
579
+ </port>
580
+ <port id="11" precision="I32">
581
+ <dim>-1</dim>
582
+ </port>
583
+ <port id="12" precision="I32">
584
+ <dim>-1</dim>
585
+ </port>
586
+ </output>
587
+ </layer>
588
+ <layer id="44" name="Constant_41106" type="Const" version="opset1">
589
+ <data element_type="i32" shape="" offset="17932002" size="4" />
590
+ <output>
591
+ <port id="0" precision="I32" />
592
+ </output>
593
+ </layer>
594
+ <layer id="45" name="Constant_41107" type="Const" version="opset1">
595
+ <data element_type="i32" shape="" offset="0" size="4" />
596
+ <output>
597
+ <port id="0" precision="I32" />
598
+ </output>
599
+ </layer>
600
+ <layer id="46" name="RaggedToDense_41108" type="RaggedToDense" version="extension">
601
+ <data pad_right="true" m_pad_max_length="false" />
602
+ <input>
603
+ <port id="0" precision="I32">
604
+ <dim>-1</dim>
605
+ </port>
606
+ <port id="1" precision="I32">
607
+ <dim>-1</dim>
608
+ </port>
609
+ <port id="2" precision="I32">
610
+ <dim>-1</dim>
611
+ </port>
612
+ <port id="3" precision="I32" />
613
+ <port id="4" precision="I32" />
614
+ </input>
615
+ <output>
616
+ <port id="5" precision="I32">
617
+ <dim>-1</dim>
618
+ <dim>200</dim>
619
+ </port>
620
+ <port id="6" precision="BOOL">
621
+ <dim>-1</dim>
622
+ <dim>200</dim>
623
+ </port>
624
+ </output>
625
+ </layer>
626
+ <layer id="47" name="Convert_41109" type="Convert" version="opset1">
627
+ <data destination_type="i32" />
628
+ <input>
629
+ <port id="0" precision="BOOL">
630
+ <dim>-1</dim>
631
+ <dim>200</dim>
632
+ </port>
633
+ </input>
634
+ <output>
635
+ <port id="1" precision="I32">
636
+ <dim>-1</dim>
637
+ <dim>200</dim>
638
+ </port>
639
+ </output>
640
+ </layer>
641
+ <layer id="48" name="Convert_41109.0" type="Convert" version="opset1">
642
+ <data destination_type="i64" />
643
+ <input>
644
+ <port id="0" precision="I32">
645
+ <dim>-1</dim>
646
+ <dim>200</dim>
647
+ </port>
648
+ </input>
649
+ <output>
650
+ <port id="1" precision="I64" names="attention_mask">
651
+ <dim>-1</dim>
652
+ <dim>200</dim>
653
+ </port>
654
+ </output>
655
+ </layer>
656
+ <layer id="49" name="RaggedToDense_41108.0" type="Convert" version="opset1">
657
+ <data destination_type="i64" />
658
+ <input>
659
+ <port id="0" precision="I32">
660
+ <dim>-1</dim>
661
+ <dim>200</dim>
662
+ </port>
663
+ </input>
664
+ <output>
665
+ <port id="1" precision="I64" names="input_ids">
666
+ <dim>-1</dim>
667
+ <dim>200</dim>
668
+ </port>
669
+ </output>
670
+ </layer>
671
+ <layer id="50" name="Result_41112" type="Result" version="opset1" output_names="input_ids">
672
+ <input>
673
+ <port id="0" precision="I64">
674
+ <dim>-1</dim>
675
+ <dim>200</dim>
676
+ </port>
677
+ </input>
678
+ </layer>
679
+ <layer id="51" name="Result_41114" type="Result" version="opset1" output_names="attention_mask">
680
+ <input>
681
+ <port id="0" precision="I64">
682
+ <dim>-1</dim>
683
+ <dim>200</dim>
684
+ </port>
685
+ </input>
686
+ </layer>
687
+ </layers>
688
+ <edges>
689
+ <edge from-layer="0" from-port="0" to-layer="5" to-port="0" />
690
+ <edge from-layer="1" from-port="0" to-layer="43" to-port="0" />
691
+ <edge from-layer="2" from-port="0" to-layer="43" to-port="1" />
692
+ <edge from-layer="3" from-port="0" to-layer="43" to-port="2" />
693
+ <edge from-layer="4" from-port="0" to-layer="11" to-port="0" />
694
+ <edge from-layer="5" from-port="1" to-layer="6" to-port="0" />
695
+ <edge from-layer="5" from-port="3" to-layer="18" to-port="4" />
696
+ <edge from-layer="5" from-port="2" to-layer="18" to-port="3" />
697
+ <edge from-layer="5" from-port="1" to-layer="18" to-port="2" />
698
+ <edge from-layer="6" from-port="1" to-layer="9" to-port="0" />
699
+ <edge from-layer="7" from-port="0" to-layer="9" to-port="1" />
700
+ <edge from-layer="8" from-port="0" to-layer="9" to-port="2" />
701
+ <edge from-layer="9" from-port="3" to-layer="14" to-port="0" />
702
+ <edge from-layer="9" from-port="3" to-layer="11" to-port="1" />
703
+ <edge from-layer="10" from-port="0" to-layer="11" to-port="2" />
704
+ <edge from-layer="11" from-port="3" to-layer="18" to-port="0" />
705
+ <edge from-layer="12" from-port="0" to-layer="16" to-port="0" />
706
+ <edge from-layer="13" from-port="0" to-layer="14" to-port="1" />
707
+ <edge from-layer="14" from-port="2" to-layer="16" to-port="1" />
708
+ <edge from-layer="15" from-port="0" to-layer="16" to-port="2" />
709
+ <edge from-layer="16" from-port="3" to-layer="18" to-port="1" />
710
+ <edge from-layer="17" from-port="0" to-layer="18" to-port="5" />
711
+ <edge from-layer="18" from-port="11" to-layer="21" to-port="3" />
712
+ <edge from-layer="18" from-port="7" to-layer="23" to-port="1" />
713
+ <edge from-layer="18" from-port="6" to-layer="23" to-port="0" />
714
+ <edge from-layer="18" from-port="10" to-layer="21" to-port="2" />
715
+ <edge from-layer="18" from-port="9" to-layer="21" to-port="1" />
716
+ <edge from-layer="18" from-port="8" to-layer="21" to-port="0" />
717
+ <edge from-layer="19" from-port="0" to-layer="21" to-port="4" />
718
+ <edge from-layer="20" from-port="0" to-layer="21" to-port="5" />
719
+ <edge from-layer="21" from-port="6" to-layer="23" to-port="2" />
720
+ <edge from-layer="21" from-port="7" to-layer="23" to-port="3" />
721
+ <edge from-layer="21" from-port="8" to-layer="23" to-port="4" />
722
+ <edge from-layer="21" from-port="9" to-layer="23" to-port="5" />
723
+ <edge from-layer="22" from-port="0" to-layer="23" to-port="6" />
724
+ <edge from-layer="23" from-port="11" to-layer="37" to-port="4" />
725
+ <edge from-layer="23" from-port="8" to-layer="37" to-port="1" />
726
+ <edge from-layer="23" from-port="9" to-layer="37" to-port="2" />
727
+ <edge from-layer="23" from-port="7" to-layer="37" to-port="0" />
728
+ <edge from-layer="23" from-port="10" to-layer="37" to-port="3" />
729
+ <edge from-layer="24" from-port="0" to-layer="37" to-port="5" />
730
+ <edge from-layer="25" from-port="0" to-layer="37" to-port="6" />
731
+ <edge from-layer="26" from-port="0" to-layer="37" to-port="7" />
732
+ <edge from-layer="27" from-port="0" to-layer="37" to-port="8" />
733
+ <edge from-layer="28" from-port="0" to-layer="37" to-port="9" />
734
+ <edge from-layer="29" from-port="0" to-layer="37" to-port="10" />
735
+ <edge from-layer="30" from-port="0" to-layer="37" to-port="11" />
736
+ <edge from-layer="31" from-port="0" to-layer="37" to-port="12" />
737
+ <edge from-layer="32" from-port="0" to-layer="37" to-port="13" />
738
+ <edge from-layer="33" from-port="0" to-layer="37" to-port="14" />
739
+ <edge from-layer="34" from-port="0" to-layer="37" to-port="15" />
740
+ <edge from-layer="35" from-port="0" to-layer="37" to-port="16" />
741
+ <edge from-layer="36" from-port="0" to-layer="37" to-port="17" />
742
+ <edge from-layer="37" from-port="20" to-layer="41" to-port="2" />
743
+ <edge from-layer="37" from-port="19" to-layer="41" to-port="1" />
744
+ <edge from-layer="37" from-port="18" to-layer="41" to-port="0" />
745
+ <edge from-layer="38" from-port="0" to-layer="41" to-port="3" />
746
+ <edge from-layer="39" from-port="0" to-layer="41" to-port="4" />
747
+ <edge from-layer="40" from-port="0" to-layer="41" to-port="5" />
748
+ <edge from-layer="41" from-port="6" to-layer="43" to-port="3" />
749
+ <edge from-layer="41" from-port="7" to-layer="43" to-port="4" />
750
+ <edge from-layer="41" from-port="8" to-layer="43" to-port="5" />
751
+ <edge from-layer="42" from-port="0" to-layer="43" to-port="6" />
752
+ <edge from-layer="43" from-port="7" to-layer="46" to-port="0" />
753
+ <edge from-layer="43" from-port="8" to-layer="46" to-port="1" />
754
+ <edge from-layer="43" from-port="9" to-layer="46" to-port="2" />
755
+ <edge from-layer="44" from-port="0" to-layer="46" to-port="3" />
756
+ <edge from-layer="45" from-port="0" to-layer="46" to-port="4" />
757
+ <edge from-layer="46" from-port="6" to-layer="47" to-port="0" />
758
+ <edge from-layer="46" from-port="5" to-layer="49" to-port="0" />
759
+ <edge from-layer="47" from-port="1" to-layer="48" to-port="0" />
760
+ <edge from-layer="48" from-port="1" to-layer="51" to-port="0" />
761
+ <edge from-layer="49" from-port="1" to-layer="50" to-port="0" />
762
+ </edges>
763
+ <rt_info>
764
+ <add_attention_mask value="True" />
765
+ <add_prefix_space />
766
+ <add_special_tokens value="True" />
767
+ <bos_token_id value="2" />
768
+ <clean_up_tokenization_spaces />
769
+ <detokenizer_input_type value="i64" />
770
+ <eos_token_id value="1" />
771
+ <handle_special_tokens_with_re />
772
+ <max_length value="200" />
773
+ <number_of_inputs value="1" />
774
+ <openvino_tokenizers_version value="2026.1.0.0-657-742d15e4be9" />
775
+ <openvino_version value="2026.1.0-21367-63e31528c62-releases/2026/1" />
776
+ <original_post_processor_template value="{&quot;type&quot;: &quot;TemplateProcessing&quot;, &quot;single&quot;: [{&quot;SpecialToken&quot;: {&quot;id&quot;: &quot;&lt;bos>&quot;, &quot;type_id&quot;: 0}}, {&quot;Sequence&quot;: {&quot;id&quot;: &quot;A&quot;, &quot;type_id&quot;: 0}}], &quot;pair&quot;: [{&quot;SpecialToken&quot;: {&quot;id&quot;: &quot;&lt;bos>&quot;, &quot;type_id&quot;: 0}}, {&quot;Sequence&quot;: {&quot;id&quot;: &quot;A&quot;, &quot;type_id&quot;: 0}}, {&quot;SpecialToken&quot;: {&quot;id&quot;: &quot;&lt;bos>&quot;, &quot;type_id&quot;: 1}}, {&quot;Sequence&quot;: {&quot;id&quot;: &quot;B&quot;, &quot;type_id&quot;: 1}}], &quot;special_tokens&quot;: {&quot;&lt;bos>&quot;: {&quot;id&quot;: &quot;&lt;bos>&quot;, &quot;ids&quot;: [2], &quot;tokens&quot;: [&quot;&lt;bos>&quot;]}}}" />
777
+ <original_tokenizer_class value="&lt;class 'transformers.models.gemma.tokenization_gemma.GemmaTokenizer'>" />
778
+ <pad_token_id value="0" />
779
+ <processed_post_processor_template value="{&quot;single&quot;: {&quot;ids&quot;: [2, -1], &quot;type_ids&quot;: [0, 0]}, &quot;pair&quot;: {&quot;ids&quot;: [2, -1, 2, -2], &quot;type_ids&quot;: [0, 0, 1, 1]}}" />
780
+ <skip_special_tokens value="True" />
781
+ <streaming_detokenizer value="False" />
782
+ <tokenizer_output_type value="i64" />
783
+ <tokenizers_version value="0.22.2" />
784
+ <transformers_version value="5.5.4" />
785
+ <use_max_padding value="True" />
786
+ <use_sentencepiece_backend value="False" />
787
+ <utf8_replace_mode value="replace" />
788
+ <with_detokenizer value="False" />
789
+ </rt_info>
790
+ </net>
phase2_snapflow/snapflow-epoch001/exports/openvino/manifest.json ADDED
@@ -0,0 +1,218 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "policy": {
3
+ "name": "pi05",
4
+ "source": {
5
+ "class_path": "physicalai.policies.pi05.policy.Pi05"
6
+ }
7
+ },
8
+ "model": {
9
+ "runner": {
10
+ "class_path": "physicalai.inference.runners.single_pass.SinglePass"
11
+ },
12
+ "artifacts": {
13
+ "openvino": "pi05.xml"
14
+ },
15
+ "preprocessors": [
16
+ {
17
+ "type": "normalize",
18
+ "stats": {
19
+ "state": {
20
+ "mean": [
21
+ 0.4902085707550287,
22
+ -15.791447722340527,
23
+ 18.025905360949306,
24
+ 55.00042447703383,
25
+ -49.21564963228982,
26
+ 9.598709224247822
27
+ ],
28
+ "std": [
29
+ 28.56218428477719,
30
+ 44.62872421028026,
31
+ 43.38945126960788,
32
+ 19.454495761579995,
33
+ 8.781436382129517,
34
+ 8.932825748525975
35
+ ],
36
+ "min": [
37
+ -71.17327117919922,
38
+ -99.66344451904297,
39
+ -56.68328094482422,
40
+ -6.712270259857178,
41
+ -85.72921752929688,
42
+ 0.26385223865509033
43
+ ],
44
+ "max": [
45
+ 80.10877990722656,
46
+ 73.74842071533203,
47
+ 98.82192993164062,
48
+ 92.56092071533203,
49
+ -3.136435031890869,
50
+ 42.084434509277344
51
+ ],
52
+ "q01": [
53
+ -39.45999045040392,
54
+ -98.81794802740971,
55
+ -23.48391958469011,
56
+ 25.50333895589101,
57
+ -61.34767717791836,
58
+ 0.6040729837518146
59
+ ],
60
+ "q99": [
61
+ 34.18319318983752,
62
+ 39.30228886401892,
63
+ 98.56118793779315,
64
+ 82.42731621305099,
65
+ -36.652074374338476,
66
+ 26.018496216201445
67
+ ],
68
+ "type": "STATE",
69
+ "name": "state",
70
+ "shape": [
71
+ 6
72
+ ]
73
+ }
74
+ },
75
+ "mode": "quantiles"
76
+ },
77
+ {
78
+ "type": "pi05",
79
+ "image_resolution": [
80
+ 224,
81
+ 224
82
+ ],
83
+ "empty_cameras": 0
84
+ },
85
+ {
86
+ "type": "ov_tokenizer",
87
+ "artifact": "tokenizer.xml"
88
+ }
89
+ ],
90
+ "postprocessors": [
91
+ {
92
+ "type": "denormalize",
93
+ "stats": {
94
+ "action": {
95
+ "mean": [
96
+ 0.47757554785419337,
97
+ -18.034652611064505,
98
+ 15.59805635303217,
99
+ 54.54463619743119,
100
+ -49.21284246567293,
101
+ 7.302610387584375
102
+ ],
103
+ "std": [
104
+ 29.091115680540447,
105
+ 44.16383911983223,
106
+ 44.81775979436524,
107
+ 19.76430732442855,
108
+ 8.799572987673075,
109
+ 10.41276650889744
110
+ ],
111
+ "min": [
112
+ -75.1649169921875,
113
+ -100.0,
114
+ -61.050724029541016,
115
+ -6.849315166473389,
116
+ -86.33736419677734,
117
+ 0.0
118
+ ],
119
+ "max": [
120
+ 81.91695404052734,
121
+ 72.79933166503906,
122
+ 100.0,
123
+ 93.32191467285156,
124
+ -2.732527494430542,
125
+ 42.88956069946289
126
+ ],
127
+ "q01": [
128
+ -41.29065982641405,
129
+ -100.0,
130
+ -27.74406375139188,
131
+ 24.550995379247542,
132
+ -61.591176330648565,
133
+ -1.000000013351432e-10
134
+ ],
135
+ "q99": [
136
+ 34.99315845909429,
137
+ 38.49093164574106,
138
+ 99.99756689137342,
139
+ 82.60630455329574,
140
+ -36.44491567201352,
141
+ 26.6719993779817
142
+ ],
143
+ "type": "ACTION",
144
+ "name": "action",
145
+ "shape": [
146
+ 6
147
+ ]
148
+ }
149
+ },
150
+ "mode": "quantiles"
151
+ }
152
+ ],
153
+ "input_features": [
154
+ {
155
+ "class_path": "physicalai.inference.data.features.InferenceFeature",
156
+ "init_args": {
157
+ "ftype": "STATE",
158
+ "shape": [
159
+ 6
160
+ ],
161
+ "name": "state",
162
+ "dtype": "float32"
163
+ }
164
+ },
165
+ {
166
+ "class_path": "physicalai.inference.data.features.InferenceFeature",
167
+ "init_args": {
168
+ "ftype": "VISUAL",
169
+ "shape": [
170
+ 3,
171
+ 480,
172
+ 640
173
+ ],
174
+ "name": "images.gripper",
175
+ "dtype": "float32"
176
+ }
177
+ },
178
+ {
179
+ "class_path": "physicalai.inference.data.features.InferenceFeature",
180
+ "init_args": {
181
+ "ftype": "VISUAL",
182
+ "shape": [
183
+ 3,
184
+ 480,
185
+ 640
186
+ ],
187
+ "name": "images.overhead",
188
+ "dtype": "float32"
189
+ }
190
+ },
191
+ {
192
+ "class_path": "physicalai.inference.data.features.InferenceFeature",
193
+ "init_args": {
194
+ "ftype": "LANGUAGE",
195
+ "shape": [],
196
+ "name": "task",
197
+ "dtype": "string"
198
+ }
199
+ }
200
+ ],
201
+ "output_features": [
202
+ {
203
+ "class_path": "physicalai.inference.data.features.InferenceFeature",
204
+ "init_args": {
205
+ "ftype": "ACTION",
206
+ "shape": [
207
+ 50,
208
+ 6
209
+ ],
210
+ "name": "action",
211
+ "dtype": "float32"
212
+ }
213
+ }
214
+ ]
215
+ },
216
+ "format": "policy_package",
217
+ "version": "1.0"
218
+ }
phase2_snapflow/snapflow-epoch001/exports/openvino/pi05.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3418fbcb1ff70c217d9003ddb2659927a9b7111658f2072daef0d565b211d803
3
+ size 6252282268
phase2_snapflow/snapflow-epoch001/exports/openvino/pi05.xml ADDED
The diff for this file is too large to render. See raw diff
 
phase2_snapflow/snapflow-epoch001/exports/openvino/tokenizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:817945e2aced98317970c4c588906f3b702ea2ace8300540e897274881a8a6a8
3
+ size 17932006
phase2_snapflow/snapflow-epoch001/exports/openvino/tokenizer.xml ADDED
@@ -0,0 +1,790 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ <?xml version="1.0"?>
2
+ <net name="tokenizer" version="11">
3
+ <layers>
4
+ <layer id="0" name="Parameter_82128" type="Parameter" version="opset1">
5
+ <data shape="?" element_type="string" />
6
+ <output>
7
+ <port id="0" precision="STRING" names="Parameter_82128">
8
+ <dim>-1</dim>
9
+ </port>
10
+ </output>
11
+ </layer>
12
+ <layer id="1" name="Constant_82247" type="Const" version="opset1">
13
+ <data element_type="i32" shape="" offset="0" size="4" />
14
+ <output>
15
+ <port id="0" precision="I32" />
16
+ </output>
17
+ </layer>
18
+ <layer id="2" name="Constant_82248" type="Const" version="opset1">
19
+ <data element_type="i32" shape="" offset="4" size="4" />
20
+ <output>
21
+ <port id="0" precision="I32" />
22
+ </output>
23
+ </layer>
24
+ <layer id="3" name="Constant_82249" type="Const" version="opset1">
25
+ <data element_type="i32" shape="1" offset="8" size="4" />
26
+ <output>
27
+ <port id="0" precision="I32">
28
+ <dim>1</dim>
29
+ </port>
30
+ </output>
31
+ </layer>
32
+ <layer id="4" name="Constant_82134" type="Const" version="opset1">
33
+ <data element_type="i64" shape="" offset="12" size="8" />
34
+ <output>
35
+ <port id="0" precision="I64" />
36
+ </output>
37
+ </layer>
38
+ <layer id="5" name="StringTensorUnpack_82129" type="StringTensorUnpack" version="opset15">
39
+ <input>
40
+ <port id="0" precision="STRING">
41
+ <dim>-1</dim>
42
+ </port>
43
+ </input>
44
+ <output>
45
+ <port id="1" precision="I32">
46
+ <dim>-1</dim>
47
+ </port>
48
+ <port id="2" precision="I32">
49
+ <dim>-1</dim>
50
+ </port>
51
+ <port id="3" precision="U8">
52
+ <dim>-1</dim>
53
+ </port>
54
+ </output>
55
+ </layer>
56
+ <layer id="6" name="ShapeOf_82130" type="ShapeOf" version="opset3">
57
+ <data output_type="i64" />
58
+ <input>
59
+ <port id="0" precision="I32">
60
+ <dim>-1</dim>
61
+ </port>
62
+ </input>
63
+ <output>
64
+ <port id="1" precision="I64">
65
+ <dim>1</dim>
66
+ </port>
67
+ </output>
68
+ </layer>
69
+ <layer id="7" name="Constant_82131" type="Const" version="opset1">
70
+ <data element_type="i64" shape="" offset="12" size="8" />
71
+ <output>
72
+ <port id="0" precision="I64" />
73
+ </output>
74
+ </layer>
75
+ <layer id="8" name="Constant_82132" type="Const" version="opset1">
76
+ <data element_type="i64" shape="" offset="12" size="8" />
77
+ <output>
78
+ <port id="0" precision="I64" />
79
+ </output>
80
+ </layer>
81
+ <layer id="9" name="Gather_82133" type="Gather" version="opset8">
82
+ <data batch_dims="0" />
83
+ <input>
84
+ <port id="0" precision="I64">
85
+ <dim>1</dim>
86
+ </port>
87
+ <port id="1" precision="I64" />
88
+ <port id="2" precision="I64" />
89
+ </input>
90
+ <output>
91
+ <port id="3" precision="I64" />
92
+ </output>
93
+ </layer>
94
+ <layer id="10" name="Constant_82135" type="Const" version="opset1">
95
+ <data element_type="i64" shape="" offset="20" size="8" />
96
+ <output>
97
+ <port id="0" precision="I64" />
98
+ </output>
99
+ </layer>
100
+ <layer id="11" name="Range_82136" type="Range" version="opset4">
101
+ <data output_type="i32" />
102
+ <input>
103
+ <port id="0" precision="I64" />
104
+ <port id="1" precision="I64" />
105
+ <port id="2" precision="I64" />
106
+ </input>
107
+ <output>
108
+ <port id="3" precision="I32">
109
+ <dim>-1</dim>
110
+ </port>
111
+ </output>
112
+ </layer>
113
+ <layer id="12" name="Constant_82137" type="Const" version="opset1">
114
+ <data element_type="i64" shape="" offset="20" size="8" />
115
+ <output>
116
+ <port id="0" precision="I64" />
117
+ </output>
118
+ </layer>
119
+ <layer id="13" name="Constant_82138" type="Const" version="opset1">
120
+ <data element_type="i64" shape="" offset="20" size="8" />
121
+ <output>
122
+ <port id="0" precision="I64" />
123
+ </output>
124
+ </layer>
125
+ <layer id="14" name="Add_82139" type="Add" version="opset1">
126
+ <data auto_broadcast="numpy" />
127
+ <input>
128
+ <port id="0" precision="I64" />
129
+ <port id="1" precision="I64" />
130
+ </input>
131
+ <output>
132
+ <port id="2" precision="I64" />
133
+ </output>
134
+ </layer>
135
+ <layer id="15" name="Constant_82140" type="Const" version="opset1">
136
+ <data element_type="i64" shape="" offset="20" size="8" />
137
+ <output>
138
+ <port id="0" precision="I64" />
139
+ </output>
140
+ </layer>
141
+ <layer id="16" name="Range_82141" type="Range" version="opset4">
142
+ <data output_type="i32" />
143
+ <input>
144
+ <port id="0" precision="I64" />
145
+ <port id="1" precision="I64" />
146
+ <port id="2" precision="I64" />
147
+ </input>
148
+ <output>
149
+ <port id="3" precision="I32">
150
+ <dim>-1</dim>
151
+ </port>
152
+ </output>
153
+ </layer>
154
+ <layer id="17" name="Constant_82205" type="Const" version="opset1">
155
+ <data element_type="u8" shape="4365" offset="28" size="4365" />
156
+ <output>
157
+ <port id="0" precision="U8">
158
+ <dim>4365</dim>
159
+ </port>
160
+ </output>
161
+ </layer>
162
+ <layer id="18" name="SpecialTokensSplit_82206" type="SpecialTokensSplit" version="extension">
163
+ <input>
164
+ <port id="0" precision="I32">
165
+ <dim>-1</dim>
166
+ </port>
167
+ <port id="1" precision="I32">
168
+ <dim>-1</dim>
169
+ </port>
170
+ <port id="2" precision="I32">
171
+ <dim>-1</dim>
172
+ </port>
173
+ <port id="3" precision="I32">
174
+ <dim>-1</dim>
175
+ </port>
176
+ <port id="4" precision="U8">
177
+ <dim>-1</dim>
178
+ </port>
179
+ <port id="5" precision="U8">
180
+ <dim>4365</dim>
181
+ </port>
182
+ </input>
183
+ <output>
184
+ <port id="6" precision="I32">
185
+ <dim>-1</dim>
186
+ </port>
187
+ <port id="7" precision="I32">
188
+ <dim>-1</dim>
189
+ </port>
190
+ <port id="8" precision="I32">
191
+ <dim>-1</dim>
192
+ </port>
193
+ <port id="9" precision="I32">
194
+ <dim>-1</dim>
195
+ </port>
196
+ <port id="10" precision="U8">
197
+ <dim>-1</dim>
198
+ </port>
199
+ <port id="11" precision="BOOL">
200
+ <dim>-1</dim>
201
+ </port>
202
+ </output>
203
+ </layer>
204
+ <layer id="19" name="Constant_82208" type="Const" version="opset1">
205
+ <data element_type="u8" shape="1" offset="4393" size="1" />
206
+ <output>
207
+ <port id="0" precision="U8">
208
+ <dim>1</dim>
209
+ </port>
210
+ </output>
211
+ </layer>
212
+ <layer id="20" name="Constant_82210" type="Const" version="opset1">
213
+ <data element_type="u8" shape="3" offset="4394" size="3" />
214
+ <output>
215
+ <port id="0" precision="U8">
216
+ <dim>3</dim>
217
+ </port>
218
+ </output>
219
+ </layer>
220
+ <layer id="21" name="RegexNormalization_82211" type="RegexNormalization" version="extension">
221
+ <data global_replace="true" />
222
+ <input>
223
+ <port id="0" precision="I32">
224
+ <dim>-1</dim>
225
+ </port>
226
+ <port id="1" precision="I32">
227
+ <dim>-1</dim>
228
+ </port>
229
+ <port id="2" precision="U8">
230
+ <dim>-1</dim>
231
+ </port>
232
+ <port id="3" precision="BOOL">
233
+ <dim>-1</dim>
234
+ </port>
235
+ <port id="4" precision="U8">
236
+ <dim>1</dim>
237
+ </port>
238
+ <port id="5" precision="U8">
239
+ <dim>3</dim>
240
+ </port>
241
+ </input>
242
+ <output>
243
+ <port id="6" precision="I32">
244
+ <dim>-1</dim>
245
+ </port>
246
+ <port id="7" precision="I32">
247
+ <dim>-1</dim>
248
+ </port>
249
+ <port id="8" precision="U8">
250
+ <dim>-1</dim>
251
+ </port>
252
+ <port id="9" precision="BOOL">
253
+ <dim>-1</dim>
254
+ </port>
255
+ </output>
256
+ </layer>
257
+ <layer id="22" name="Constant_82213" type="Const" version="opset1">
258
+ <data element_type="u8" shape="1" offset="4393" size="1" />
259
+ <output>
260
+ <port id="0" precision="U8">
261
+ <dim>1</dim>
262
+ </port>
263
+ </output>
264
+ </layer>
265
+ <layer id="23" name="RegexSplit_82214" type="RegexSplit" version="extension">
266
+ <data behaviour="mergedwithprevious" invert="false" max_splits="-1" />
267
+ <input>
268
+ <port id="0" precision="I32">
269
+ <dim>-1</dim>
270
+ </port>
271
+ <port id="1" precision="I32">
272
+ <dim>-1</dim>
273
+ </port>
274
+ <port id="2" precision="I32">
275
+ <dim>-1</dim>
276
+ </port>
277
+ <port id="3" precision="I32">
278
+ <dim>-1</dim>
279
+ </port>
280
+ <port id="4" precision="U8">
281
+ <dim>-1</dim>
282
+ </port>
283
+ <port id="5" precision="BOOL">
284
+ <dim>-1</dim>
285
+ </port>
286
+ <port id="6" precision="U8">
287
+ <dim>1</dim>
288
+ </port>
289
+ </input>
290
+ <output>
291
+ <port id="7" precision="I32">
292
+ <dim>-1</dim>
293
+ </port>
294
+ <port id="8" precision="I32">
295
+ <dim>-1</dim>
296
+ </port>
297
+ <port id="9" precision="I32">
298
+ <dim>-1</dim>
299
+ </port>
300
+ <port id="10" precision="I32">
301
+ <dim>-1</dim>
302
+ </port>
303
+ <port id="11" precision="U8">
304
+ <dim>-1</dim>
305
+ </port>
306
+ <port id="12" precision="BOOL">
307
+ <dim>-1</dim>
308
+ </port>
309
+ </output>
310
+ </layer>
311
+ <layer id="24" name="Constant_82216" type="Const" version="opset1">
312
+ <data element_type="i32" shape="257153" offset="4397" size="1028612" />
313
+ <output>
314
+ <port id="0" precision="I32">
315
+ <dim>257153</dim>
316
+ </port>
317
+ </output>
318
+ </layer>
319
+ <layer id="25" name="Constant_82218" type="Const" version="opset1">
320
+ <data element_type="i32" shape="257153" offset="1033009" size="1028612" />
321
+ <output>
322
+ <port id="0" precision="I32">
323
+ <dim>257153</dim>
324
+ </port>
325
+ </output>
326
+ </layer>
327
+ <layer id="26" name="Constant_82220" type="Const" version="opset1">
328
+ <data element_type="u8" shape="1942155" offset="2061621" size="1942155" />
329
+ <output>
330
+ <port id="0" precision="U8">
331
+ <dim>1942155</dim>
332
+ </port>
333
+ </output>
334
+ </layer>
335
+ <layer id="27" name="Constant_82228" type="Const" version="opset1">
336
+ <data element_type="i32" shape="580604" offset="4003776" size="2322416" />
337
+ <output>
338
+ <port id="0" precision="I32">
339
+ <dim>580604</dim>
340
+ </port>
341
+ </output>
342
+ </layer>
343
+ <layer id="28" name="Constant_82230" type="Const" version="opset1">
344
+ <data element_type="i32" shape="580604" offset="6326192" size="2322416" />
345
+ <output>
346
+ <port id="0" precision="I32">
347
+ <dim>580604</dim>
348
+ </port>
349
+ </output>
350
+ </layer>
351
+ <layer id="29" name="Constant_82232" type="Const" version="opset1">
352
+ <data element_type="u8" shape="2709312" offset="8648608" size="2709312" />
353
+ <output>
354
+ <port id="0" precision="U8">
355
+ <dim>2709312</dim>
356
+ </port>
357
+ </output>
358
+ </layer>
359
+ <layer id="30" name="Constant_82234" type="Const" version="opset1">
360
+ <data element_type="i32" shape="580604" offset="11357920" size="2322416" />
361
+ <output>
362
+ <port id="0" precision="I32">
363
+ <dim>580604</dim>
364
+ </port>
365
+ </output>
366
+ </layer>
367
+ <layer id="31" name="Constant_82236" type="Const" version="opset1">
368
+ <data element_type="i32" shape="580604" offset="13680336" size="2322416" />
369
+ <output>
370
+ <port id="0" precision="I32">
371
+ <dim>580604</dim>
372
+ </port>
373
+ </output>
374
+ </layer>
375
+ <layer id="32" name="Constant_82238" type="Const" version="opset1">
376
+ <data element_type="u8" shape="1923319" offset="16002752" size="1923319" />
377
+ <output>
378
+ <port id="0" precision="U8">
379
+ <dim>1923319</dim>
380
+ </port>
381
+ </output>
382
+ </layer>
383
+ <layer id="33" name="Constant_82222" type="Const" version="opset1">
384
+ <data element_type="i32" shape="217" offset="17926071" size="868" />
385
+ <output>
386
+ <port id="0" precision="I32">
387
+ <dim>217</dim>
388
+ </port>
389
+ </output>
390
+ </layer>
391
+ <layer id="34" name="Constant_82224" type="Const" version="opset1">
392
+ <data element_type="i32" shape="217" offset="17926939" size="868" />
393
+ <output>
394
+ <port id="0" precision="I32">
395
+ <dim>217</dim>
396
+ </port>
397
+ </output>
398
+ </layer>
399
+ <layer id="35" name="Constant_82226" type="Const" version="opset1">
400
+ <data element_type="u8" shape="3305" offset="17927807" size="3305" />
401
+ <output>
402
+ <port id="0" precision="U8">
403
+ <dim>3305</dim>
404
+ </port>
405
+ </output>
406
+ </layer>
407
+ <layer id="36" name="Constant_82239" type="Const" version="opset1">
408
+ <data element_type="i32" shape="217" offset="17931112" size="868" />
409
+ <output>
410
+ <port id="0" precision="I32">
411
+ <dim>217</dim>
412
+ </port>
413
+ </output>
414
+ </layer>
415
+ <layer id="37" name="BPETokenizer_82240" type="BPETokenizer" version="extension">
416
+ <data unk_token="&lt;unk>" fuse_unk="true" suffix_indicator="" end_suffix="" byte_fallback="true" cache_capacity="51430" />
417
+ <input>
418
+ <port id="0" precision="I32">
419
+ <dim>-1</dim>
420
+ </port>
421
+ <port id="1" precision="I32">
422
+ <dim>-1</dim>
423
+ </port>
424
+ <port id="2" precision="I32">
425
+ <dim>-1</dim>
426
+ </port>
427
+ <port id="3" precision="I32">
428
+ <dim>-1</dim>
429
+ </port>
430
+ <port id="4" precision="U8">
431
+ <dim>-1</dim>
432
+ </port>
433
+ <port id="5" precision="I32">
434
+ <dim>257153</dim>
435
+ </port>
436
+ <port id="6" precision="I32">
437
+ <dim>257153</dim>
438
+ </port>
439
+ <port id="7" precision="U8">
440
+ <dim>1942155</dim>
441
+ </port>
442
+ <port id="8" precision="I32">
443
+ <dim>580604</dim>
444
+ </port>
445
+ <port id="9" precision="I32">
446
+ <dim>580604</dim>
447
+ </port>
448
+ <port id="10" precision="U8">
449
+ <dim>2709312</dim>
450
+ </port>
451
+ <port id="11" precision="I32">
452
+ <dim>580604</dim>
453
+ </port>
454
+ <port id="12" precision="I32">
455
+ <dim>580604</dim>
456
+ </port>
457
+ <port id="13" precision="U8">
458
+ <dim>1923319</dim>
459
+ </port>
460
+ <port id="14" precision="I32">
461
+ <dim>217</dim>
462
+ </port>
463
+ <port id="15" precision="I32">
464
+ <dim>217</dim>
465
+ </port>
466
+ <port id="16" precision="U8">
467
+ <dim>3305</dim>
468
+ </port>
469
+ <port id="17" precision="I32">
470
+ <dim>217</dim>
471
+ </port>
472
+ </input>
473
+ <output>
474
+ <port id="18" precision="I32">
475
+ <dim>-1</dim>
476
+ </port>
477
+ <port id="19" precision="I32">
478
+ <dim>-1</dim>
479
+ </port>
480
+ <port id="20" precision="I32">
481
+ <dim>-1</dim>
482
+ </port>
483
+ </output>
484
+ </layer>
485
+ <layer id="38" name="Constant_82241" type="Const" version="opset1">
486
+ <data element_type="i32" shape="" offset="17931980" size="4" />
487
+ <output>
488
+ <port id="0" precision="I32" />
489
+ </output>
490
+ </layer>
491
+ <layer id="39" name="Constant_82243" type="Const" version="opset1">
492
+ <data element_type="u8" shape="5" offset="17931984" size="5" />
493
+ <output>
494
+ <port id="0" precision="U8">
495
+ <dim>5</dim>
496
+ </port>
497
+ </output>
498
+ </layer>
499
+ <layer id="40" name="Constant_82245" type="Const" version="opset1">
500
+ <data element_type="u8" shape="13" offset="17931989" size="13" />
501
+ <output>
502
+ <port id="0" precision="U8">
503
+ <dim>13</dim>
504
+ </port>
505
+ </output>
506
+ </layer>
507
+ <layer id="41" name="Truncate_82246" type="Truncate" version="extension">
508
+ <data m_num_inputs="1" />
509
+ <input>
510
+ <port id="0" precision="I32">
511
+ <dim>-1</dim>
512
+ </port>
513
+ <port id="1" precision="I32">
514
+ <dim>-1</dim>
515
+ </port>
516
+ <port id="2" precision="I32">
517
+ <dim>-1</dim>
518
+ </port>
519
+ <port id="3" precision="I32" />
520
+ <port id="4" precision="U8">
521
+ <dim>5</dim>
522
+ </port>
523
+ <port id="5" precision="U8">
524
+ <dim>13</dim>
525
+ </port>
526
+ </input>
527
+ <output>
528
+ <port id="6" precision="I32">
529
+ <dim>-1</dim>
530
+ </port>
531
+ <port id="7" precision="I32">
532
+ <dim>-1</dim>
533
+ </port>
534
+ <port id="8" precision="I32">
535
+ <dim>-1</dim>
536
+ </port>
537
+ </output>
538
+ </layer>
539
+ <layer id="42" name="Constant_82250" type="Const" version="opset1">
540
+ <data element_type="i32" shape="2" offset="12" size="8" />
541
+ <output>
542
+ <port id="0" precision="I32">
543
+ <dim>2</dim>
544
+ </port>
545
+ </output>
546
+ </layer>
547
+ <layer id="43" name="CombineSegments_82251" type="CombineSegments" version="extension">
548
+ <input>
549
+ <port id="0" precision="I32" />
550
+ <port id="1" precision="I32" />
551
+ <port id="2" precision="I32">
552
+ <dim>1</dim>
553
+ </port>
554
+ <port id="3" precision="I32">
555
+ <dim>-1</dim>
556
+ </port>
557
+ <port id="4" precision="I32">
558
+ <dim>-1</dim>
559
+ </port>
560
+ <port id="5" precision="I32">
561
+ <dim>-1</dim>
562
+ </port>
563
+ <port id="6" precision="I32">
564
+ <dim>2</dim>
565
+ </port>
566
+ </input>
567
+ <output>
568
+ <port id="7" precision="I32">
569
+ <dim>-1</dim>
570
+ </port>
571
+ <port id="8" precision="I32">
572
+ <dim>-1</dim>
573
+ </port>
574
+ <port id="9" precision="I32">
575
+ <dim>-1</dim>
576
+ </port>
577
+ <port id="10" precision="I32">
578
+ <dim>-1</dim>
579
+ </port>
580
+ <port id="11" precision="I32">
581
+ <dim>-1</dim>
582
+ </port>
583
+ <port id="12" precision="I32">
584
+ <dim>-1</dim>
585
+ </port>
586
+ </output>
587
+ </layer>
588
+ <layer id="44" name="Constant_82252" type="Const" version="opset1">
589
+ <data element_type="i32" shape="" offset="17932002" size="4" />
590
+ <output>
591
+ <port id="0" precision="I32" />
592
+ </output>
593
+ </layer>
594
+ <layer id="45" name="Constant_82253" type="Const" version="opset1">
595
+ <data element_type="i32" shape="" offset="0" size="4" />
596
+ <output>
597
+ <port id="0" precision="I32" />
598
+ </output>
599
+ </layer>
600
+ <layer id="46" name="RaggedToDense_82254" type="RaggedToDense" version="extension">
601
+ <data pad_right="true" m_pad_max_length="false" />
602
+ <input>
603
+ <port id="0" precision="I32">
604
+ <dim>-1</dim>
605
+ </port>
606
+ <port id="1" precision="I32">
607
+ <dim>-1</dim>
608
+ </port>
609
+ <port id="2" precision="I32">
610
+ <dim>-1</dim>
611
+ </port>
612
+ <port id="3" precision="I32" />
613
+ <port id="4" precision="I32" />
614
+ </input>
615
+ <output>
616
+ <port id="5" precision="I32">
617
+ <dim>-1</dim>
618
+ <dim>200</dim>
619
+ </port>
620
+ <port id="6" precision="BOOL">
621
+ <dim>-1</dim>
622
+ <dim>200</dim>
623
+ </port>
624
+ </output>
625
+ </layer>
626
+ <layer id="47" name="Convert_82255" type="Convert" version="opset1">
627
+ <data destination_type="i32" />
628
+ <input>
629
+ <port id="0" precision="BOOL">
630
+ <dim>-1</dim>
631
+ <dim>200</dim>
632
+ </port>
633
+ </input>
634
+ <output>
635
+ <port id="1" precision="I32">
636
+ <dim>-1</dim>
637
+ <dim>200</dim>
638
+ </port>
639
+ </output>
640
+ </layer>
641
+ <layer id="48" name="Convert_82255.0" type="Convert" version="opset1">
642
+ <data destination_type="i64" />
643
+ <input>
644
+ <port id="0" precision="I32">
645
+ <dim>-1</dim>
646
+ <dim>200</dim>
647
+ </port>
648
+ </input>
649
+ <output>
650
+ <port id="1" precision="I64" names="attention_mask">
651
+ <dim>-1</dim>
652
+ <dim>200</dim>
653
+ </port>
654
+ </output>
655
+ </layer>
656
+ <layer id="49" name="RaggedToDense_82254.0" type="Convert" version="opset1">
657
+ <data destination_type="i64" />
658
+ <input>
659
+ <port id="0" precision="I32">
660
+ <dim>-1</dim>
661
+ <dim>200</dim>
662
+ </port>
663
+ </input>
664
+ <output>
665
+ <port id="1" precision="I64" names="input_ids">
666
+ <dim>-1</dim>
667
+ <dim>200</dim>
668
+ </port>
669
+ </output>
670
+ </layer>
671
+ <layer id="50" name="Result_82258" type="Result" version="opset1" output_names="input_ids">
672
+ <input>
673
+ <port id="0" precision="I64">
674
+ <dim>-1</dim>
675
+ <dim>200</dim>
676
+ </port>
677
+ </input>
678
+ </layer>
679
+ <layer id="51" name="Result_82260" type="Result" version="opset1" output_names="attention_mask">
680
+ <input>
681
+ <port id="0" precision="I64">
682
+ <dim>-1</dim>
683
+ <dim>200</dim>
684
+ </port>
685
+ </input>
686
+ </layer>
687
+ </layers>
688
+ <edges>
689
+ <edge from-layer="0" from-port="0" to-layer="5" to-port="0" />
690
+ <edge from-layer="1" from-port="0" to-layer="43" to-port="0" />
691
+ <edge from-layer="2" from-port="0" to-layer="43" to-port="1" />
692
+ <edge from-layer="3" from-port="0" to-layer="43" to-port="2" />
693
+ <edge from-layer="4" from-port="0" to-layer="11" to-port="0" />
694
+ <edge from-layer="5" from-port="1" to-layer="6" to-port="0" />
695
+ <edge from-layer="5" from-port="3" to-layer="18" to-port="4" />
696
+ <edge from-layer="5" from-port="2" to-layer="18" to-port="3" />
697
+ <edge from-layer="5" from-port="1" to-layer="18" to-port="2" />
698
+ <edge from-layer="6" from-port="1" to-layer="9" to-port="0" />
699
+ <edge from-layer="7" from-port="0" to-layer="9" to-port="1" />
700
+ <edge from-layer="8" from-port="0" to-layer="9" to-port="2" />
701
+ <edge from-layer="9" from-port="3" to-layer="14" to-port="0" />
702
+ <edge from-layer="9" from-port="3" to-layer="11" to-port="1" />
703
+ <edge from-layer="10" from-port="0" to-layer="11" to-port="2" />
704
+ <edge from-layer="11" from-port="3" to-layer="18" to-port="0" />
705
+ <edge from-layer="12" from-port="0" to-layer="16" to-port="0" />
706
+ <edge from-layer="13" from-port="0" to-layer="14" to-port="1" />
707
+ <edge from-layer="14" from-port="2" to-layer="16" to-port="1" />
708
+ <edge from-layer="15" from-port="0" to-layer="16" to-port="2" />
709
+ <edge from-layer="16" from-port="3" to-layer="18" to-port="1" />
710
+ <edge from-layer="17" from-port="0" to-layer="18" to-port="5" />
711
+ <edge from-layer="18" from-port="11" to-layer="21" to-port="3" />
712
+ <edge from-layer="18" from-port="7" to-layer="23" to-port="1" />
713
+ <edge from-layer="18" from-port="6" to-layer="23" to-port="0" />
714
+ <edge from-layer="18" from-port="10" to-layer="21" to-port="2" />
715
+ <edge from-layer="18" from-port="9" to-layer="21" to-port="1" />
716
+ <edge from-layer="18" from-port="8" to-layer="21" to-port="0" />
717
+ <edge from-layer="19" from-port="0" to-layer="21" to-port="4" />
718
+ <edge from-layer="20" from-port="0" to-layer="21" to-port="5" />
719
+ <edge from-layer="21" from-port="6" to-layer="23" to-port="2" />
720
+ <edge from-layer="21" from-port="7" to-layer="23" to-port="3" />
721
+ <edge from-layer="21" from-port="8" to-layer="23" to-port="4" />
722
+ <edge from-layer="21" from-port="9" to-layer="23" to-port="5" />
723
+ <edge from-layer="22" from-port="0" to-layer="23" to-port="6" />
724
+ <edge from-layer="23" from-port="11" to-layer="37" to-port="4" />
725
+ <edge from-layer="23" from-port="8" to-layer="37" to-port="1" />
726
+ <edge from-layer="23" from-port="9" to-layer="37" to-port="2" />
727
+ <edge from-layer="23" from-port="7" to-layer="37" to-port="0" />
728
+ <edge from-layer="23" from-port="10" to-layer="37" to-port="3" />
729
+ <edge from-layer="24" from-port="0" to-layer="37" to-port="5" />
730
+ <edge from-layer="25" from-port="0" to-layer="37" to-port="6" />
731
+ <edge from-layer="26" from-port="0" to-layer="37" to-port="7" />
732
+ <edge from-layer="27" from-port="0" to-layer="37" to-port="8" />
733
+ <edge from-layer="28" from-port="0" to-layer="37" to-port="9" />
734
+ <edge from-layer="29" from-port="0" to-layer="37" to-port="10" />
735
+ <edge from-layer="30" from-port="0" to-layer="37" to-port="11" />
736
+ <edge from-layer="31" from-port="0" to-layer="37" to-port="12" />
737
+ <edge from-layer="32" from-port="0" to-layer="37" to-port="13" />
738
+ <edge from-layer="33" from-port="0" to-layer="37" to-port="14" />
739
+ <edge from-layer="34" from-port="0" to-layer="37" to-port="15" />
740
+ <edge from-layer="35" from-port="0" to-layer="37" to-port="16" />
741
+ <edge from-layer="36" from-port="0" to-layer="37" to-port="17" />
742
+ <edge from-layer="37" from-port="20" to-layer="41" to-port="2" />
743
+ <edge from-layer="37" from-port="19" to-layer="41" to-port="1" />
744
+ <edge from-layer="37" from-port="18" to-layer="41" to-port="0" />
745
+ <edge from-layer="38" from-port="0" to-layer="41" to-port="3" />
746
+ <edge from-layer="39" from-port="0" to-layer="41" to-port="4" />
747
+ <edge from-layer="40" from-port="0" to-layer="41" to-port="5" />
748
+ <edge from-layer="41" from-port="6" to-layer="43" to-port="3" />
749
+ <edge from-layer="41" from-port="7" to-layer="43" to-port="4" />
750
+ <edge from-layer="41" from-port="8" to-layer="43" to-port="5" />
751
+ <edge from-layer="42" from-port="0" to-layer="43" to-port="6" />
752
+ <edge from-layer="43" from-port="7" to-layer="46" to-port="0" />
753
+ <edge from-layer="43" from-port="8" to-layer="46" to-port="1" />
754
+ <edge from-layer="43" from-port="9" to-layer="46" to-port="2" />
755
+ <edge from-layer="44" from-port="0" to-layer="46" to-port="3" />
756
+ <edge from-layer="45" from-port="0" to-layer="46" to-port="4" />
757
+ <edge from-layer="46" from-port="6" to-layer="47" to-port="0" />
758
+ <edge from-layer="46" from-port="5" to-layer="49" to-port="0" />
759
+ <edge from-layer="47" from-port="1" to-layer="48" to-port="0" />
760
+ <edge from-layer="48" from-port="1" to-layer="51" to-port="0" />
761
+ <edge from-layer="49" from-port="1" to-layer="50" to-port="0" />
762
+ </edges>
763
+ <rt_info>
764
+ <add_attention_mask value="True" />
765
+ <add_prefix_space />
766
+ <add_special_tokens value="True" />
767
+ <bos_token_id value="2" />
768
+ <clean_up_tokenization_spaces />
769
+ <detokenizer_input_type value="i64" />
770
+ <eos_token_id value="1" />
771
+ <handle_special_tokens_with_re />
772
+ <max_length value="200" />
773
+ <number_of_inputs value="1" />
774
+ <openvino_tokenizers_version value="2026.1.0.0-657-742d15e4be9" />
775
+ <openvino_version value="2026.1.0-21367-63e31528c62-releases/2026/1" />
776
+ <original_post_processor_template value="{&quot;type&quot;: &quot;TemplateProcessing&quot;, &quot;single&quot;: [{&quot;SpecialToken&quot;: {&quot;id&quot;: &quot;&lt;bos>&quot;, &quot;type_id&quot;: 0}}, {&quot;Sequence&quot;: {&quot;id&quot;: &quot;A&quot;, &quot;type_id&quot;: 0}}], &quot;pair&quot;: [{&quot;SpecialToken&quot;: {&quot;id&quot;: &quot;&lt;bos>&quot;, &quot;type_id&quot;: 0}}, {&quot;Sequence&quot;: {&quot;id&quot;: &quot;A&quot;, &quot;type_id&quot;: 0}}, {&quot;SpecialToken&quot;: {&quot;id&quot;: &quot;&lt;bos>&quot;, &quot;type_id&quot;: 1}}, {&quot;Sequence&quot;: {&quot;id&quot;: &quot;B&quot;, &quot;type_id&quot;: 1}}], &quot;special_tokens&quot;: {&quot;&lt;bos>&quot;: {&quot;id&quot;: &quot;&lt;bos>&quot;, &quot;ids&quot;: [2], &quot;tokens&quot;: [&quot;&lt;bos>&quot;]}}}" />
777
+ <original_tokenizer_class value="&lt;class 'transformers.models.gemma.tokenization_gemma.GemmaTokenizer'>" />
778
+ <pad_token_id value="0" />
779
+ <processed_post_processor_template value="{&quot;single&quot;: {&quot;ids&quot;: [2, -1], &quot;type_ids&quot;: [0, 0]}, &quot;pair&quot;: {&quot;ids&quot;: [2, -1, 2, -2], &quot;type_ids&quot;: [0, 0, 1, 1]}}" />
780
+ <skip_special_tokens value="True" />
781
+ <streaming_detokenizer value="False" />
782
+ <tokenizer_output_type value="i64" />
783
+ <tokenizers_version value="0.22.2" />
784
+ <transformers_version value="5.5.4" />
785
+ <use_max_padding value="True" />
786
+ <use_sentencepiece_backend value="False" />
787
+ <utf8_replace_mode value="replace" />
788
+ <with_detokenizer value="False" />
789
+ </rt_info>
790
+ </net>
phase2_snapflow/wandb/debug-internal.log ADDED
The diff for this file is too large to render. See raw diff
 
phase2_snapflow/wandb/debug.log ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_setup.py:_flush():81] Current SDK version is 0.28.1
2
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_setup.py:_flush():81] Configure stats pid to 1916126
3
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_setup.py:_flush():81] Loading settings from environment variables
4
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:setup_run_log_directory():729] Logging user logs to /mnt/data/experiments/snapflow-multi-obj/phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/logs/debug.log
5
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:setup_run_log_directory():730] Logging internal logs to /mnt/data/experiments/snapflow-multi-obj/phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/logs/debug-internal.log
6
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:init():772] calling init triggers
7
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:init():777] wandb.init called with sweep_config: {}
8
+ config: {'_wandb': {}}
9
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:init():820] starting backend
10
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:init():835] sending inform_init request
11
+ 2026-08-07 09:38:05,745 INFO MainThread:1916126 [wandb_init.py:init():840] backend started and connected
12
+ 2026-08-07 09:38:05,747 INFO MainThread:1916126 [wandb_init.py:init():910] updated telemetry
13
+ 2026-08-07 09:38:05,755 INFO MainThread:1916126 [wandb_init.py:init():933] communicating run to backend with 90.0 second timeout
14
+ 2026-08-07 09:38:06,580 INFO MainThread:1916126 [wandb_init.py:init():978] starting run threads in backend
15
+ 2026-08-07 09:38:06,731 INFO MainThread:1916126 [wandb_run.py:_console_start():2621] atexit reg
16
+ 2026-08-07 09:38:06,731 INFO MainThread:1916126 [wandb_run.py:_redirect():2471] redirect: wrap_raw
17
+ 2026-08-07 09:38:06,731 INFO MainThread:1916126 [wandb_run.py:_redirect():2540] Wrapping output streams.
18
+ 2026-08-07 09:38:06,731 INFO MainThread:1916126 [wandb_run.py:_redirect():2563] Redirects installed.
19
+ 2026-08-07 09:38:06,738 INFO MainThread:1916126 [wandb_init.py:init():1016] run started, returning control to user process
20
+ 2026-08-07 09:38:06,739 INFO MainThread:1916126 [wandb_run.py:_config_callback():1346] config_cb None None {'phase': 'phase2-snapflow', 'dataset': 'Daankrol/pick-and-place-multi-obj', 'pretrained': 'lerobot/pi05_base', 'seed': 42, 'val_split': 0.1, 'val_split_seed': 42, 'augment': True, 'augment_n_subset': 3, 'batch_size': 16, 'devices': 1, 'accumulate_grad_batches': 1, 'epochs': 3, 'train_frames': 86093, 'global_batch': 16, 'steps_per_epoch': 5380, 'total_steps': 16140, 'warmup_steps': 807, 'warmup_frac': 0.05, 'snapflow_alpha': 0.5, 'snapflow_lambda': 0.1, 'snapflow_steps': 1}
21
+ 2026-08-07 09:38:09,271 INFO MainThread:1916126 [wandb_run.py:_config_callback():1346] config_cb None None {'paligemma_variant': 'gemma_2b', 'action_expert_variant': 'gemma_300m', 'dtype': 'bfloat16', 'n_obs_steps': 1, 'chunk_size': 50, 'n_action_steps': 50, 'max_state_dim': 32, 'max_action_dim': 32, 'num_inference_steps': 10, 'time_sampling_beta_alpha': 1.5, 'time_sampling_beta_beta': 1.0, 'time_sampling_scale': 0.999, 'time_sampling_offset': 0.001, 'min_period': 0.004, 'max_period': 4.0, 'use_random_input_noise': True, 'snapflow_enabled': True, 'snapflow_alpha': 0.5, 'snapflow_lambda': 0.1, 'snapflow_num_inference_steps': 1, 'image_resolution': [224, 224], 'empty_cameras': 0, 'tokenizer_max_length': 200, 'gradient_checkpointing': True, 'compile_mode': 'max-autotune', 'freeze_vision_encoder': False, 'train_expert_only': True, 'normalization_mode': 'QUANTILES', 'optimizer_lr': 2.5e-05, 'optimizer_betas': [0.9, 0.95], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 0.01, 'optimizer_grad_clip_norm': 1.0, 'scheduler_warmup_steps': 807, 'scheduler_decay_steps': None, 'scheduler_decay_lr': 2.5e-06, 'dataset_stats': {'observation.state': {'mean': [0.4902085707550287, -15.791447722340527, 18.025905360949306, 55.00042447703383, -49.21564963228982, 9.598709224247822], 'std': [28.56218428477719, 44.62872421028026, 43.38945126960788, 19.454495761579995, 8.781436382129517, 8.932825748525975], 'min': [-71.17327117919922, -99.66344451904297, -56.68328094482422, -6.712270259857178, -85.72921752929688, 0.26385223865509033], 'max': [80.10877990722656, 73.74842071533203, 98.82192993164062, 92.56092071533203, -3.136435031890869, 42.084434509277344], 'q01': [-39.45999045040392, -98.81794802740971, -23.48391958469011, 25.50333895589101, -61.34767717791836, 0.6040729837518146], 'q99': [34.18319318983752, 39.30228886401892, 98.56118793779315, 82.42731621305099, -36.652074374338476, 26.018496216201445], 'type': 'STATE', 'name': 'state', 'shape': [6]}, 'observation.gripper': {'mean': [[[0.5900507834223626]], [[0.5872656129005113]], [[0.5966057530402353]]], 'std': [[[0.1831069655022816]], [[0.18826039482586307]], [[0.18464241941191994]]], 'min': [[[0.0]], [[0.0]], [[0.0]]], 'max': [[[1.0]], [[1.0]], [[1.0]]], 'q01': [[[0.07565626654408032]], [[0.07064545037186447]], [[0.09675786695048419]]], 'q99': [[[0.8841095335313836]], [[0.9249016521748583]], [[0.9647414252015477]]], 'type': 'VISUAL', 'name': 'gripper', 'shape': [3, 480, 640]}, 'observation.overhead': {'mean': [[[0.545862979096059]], [[0.5452054301949182]], [[0.5477746602359932]]], 'std': [[[0.14197105669614393]], [[0.1436179498089634]], [[0.14500463044962006]]], 'min': [[[0.0]], [[0.0]], [[0.0]]], 'max': [[[1.0]], [[1.0]], [[1.0]]], 'q01': [[[0.05494559589188223]], [[0.05129975818181184]], [[0.07399745843495421]]], 'q99': [[[0.7951740036832365]], [[0.8620514435976284]], [[0.9287900526200447]]], 'type': 'VISUAL', 'name': 'overhead', 'shape': [3, 480, 640]}, 'action': {'mean': [0.47757554785419337, -18.034652611064505, 15.59805635303217, 54.54463619743119, -49.21284246567293, 7.302610387584375], 'std': [29.091115680540447, 44.16383911983223, 44.81775979436524, 19.76430732442855, 8.799572987673075, 10.41276650889744], 'min': [-75.1649169921875, -100.0, -61.050724029541016, -6.849315166473389, -86.33736419677734, 0.0], 'max': [81.91695404052734, 72.79933166503906, 100.0, 93.32191467285156, -2.732527494430542, 42.88956069946289], 'q01': [-41.29065982641405, -100.0, -27.74406375139188, 24.550995379247542, -61.591176330648565, -1.000000013351432e-10], 'q99': [34.99315845909429, 38.49093164574106, 99.99756689137342, 82.60630455329574, -36.44491567201352, 26.6719993779817], 'type': 'ACTION', 'name': 'action', 'shape': [6]}}}
phase2_snapflow/wandb/run-20260806_235610-358tcehl/files/config.yaml ADDED
@@ -0,0 +1,375 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ _wandb:
2
+ value:
3
+ cli_version: 0.28.1
4
+ e:
5
+ 0nwnfyvby6u9m7fidg3frxdoi5x5535q:
6
+ args:
7
+ - --wandb
8
+ - --augment
9
+ - --phase1-epochs
10
+ - "7"
11
+ - --dataset
12
+ - Daankrol/pick-and-place-multi-obj
13
+ - --output-dir
14
+ - /mnt/data/experiments/snapflow-multi-obj
15
+ codePath: library/scripts/train_snapflow_so101.py
16
+ codePathLocal: library/scripts/train_snapflow_so101.py
17
+ cpu_count: 64
18
+ cpu_count_logical: 128
19
+ cudaVersion: "13.2"
20
+ disk:
21
+ /:
22
+ total: "1887976378368"
23
+ used: "1660689944576"
24
+ email: daankrol@hotmail.com
25
+ executable: /home/devuser/daankrol/physical-ai-studio/library/.venv/bin/python
26
+ git:
27
+ commit: f3a674359c427756b3abf7fe50766f03bc3da2d4
28
+ remote: https://github.com/open-edge-platform/physical-ai-studio.git
29
+ gpu: NVIDIA H100 PCIe
30
+ gpu_count: 4
31
+ gpu_nvidia:
32
+ - architecture: Hopper
33
+ cudaCores: 14592
34
+ memoryTotal: "85520809984"
35
+ name: NVIDIA H100 PCIe
36
+ uuid: GPU-99931a32-3903-4dd2-57aa-3f5a87cceab6
37
+ - architecture: Hopper
38
+ cudaCores: 14592
39
+ memoryTotal: "85520809984"
40
+ name: NVIDIA H100 PCIe
41
+ uuid: GPU-48a80548-2790-2ee9-d239-752ba918214e
42
+ - architecture: Hopper
43
+ cudaCores: 14592
44
+ memoryTotal: "85520809984"
45
+ name: NVIDIA H100 PCIe
46
+ uuid: GPU-dcc57cb5-b02f-ab7f-751d-4d55df91356d
47
+ - architecture: Hopper
48
+ cudaCores: 14592
49
+ memoryTotal: "85520809984"
50
+ name: NVIDIA H100 PCIe
51
+ uuid: GPU-73153eba-ee9b-1bd6-c1cc-2f14527d0497
52
+ host: ikvensx013
53
+ memory:
54
+ total: "1352432504832"
55
+ os: Linux-6.8.0-136-generic-x86_64-with-glibc2.39
56
+ program: /home/devuser/daankrol/physical-ai-studio/library/scripts/train_snapflow_so101.py
57
+ python: CPython 3.12.3
58
+ root: /mnt/data/experiments/snapflow-multi-obj/phase2_snapflow
59
+ startedAt: "2026-08-06T23:56:10.250950Z"
60
+ writerId: 0nwnfyvby6u9m7fidg3frxdoi5x5535q
61
+ m:
62
+ - "1": trainer/global_step
63
+ "6":
64
+ - 3
65
+ "7": []
66
+ - "2": '*'
67
+ "5": 1
68
+ "6":
69
+ - 1
70
+ "7": []
71
+ python_version: 3.12.3
72
+ t:
73
+ "1":
74
+ - 1
75
+ - 11
76
+ - 41
77
+ - 49
78
+ - 51
79
+ - 71
80
+ - 83
81
+ - 98
82
+ - 106
83
+ "2":
84
+ - 1
85
+ - 11
86
+ - 41
87
+ - 49
88
+ - 51
89
+ - 71
90
+ - 83
91
+ - 98
92
+ - 106
93
+ "3":
94
+ - 2
95
+ - 7
96
+ - 13
97
+ - 15
98
+ - 66
99
+ "4": 3.12.3
100
+ "5": 0.28.1
101
+ "6": 5.5.4
102
+ "12": 0.28.1
103
+ "13": linux-x86_64
104
+ accumulate_grad_batches:
105
+ value: 1
106
+ action_expert_variant:
107
+ value: gemma_300m
108
+ augment:
109
+ value: true
110
+ augment_n_subset:
111
+ value: 3
112
+ batch_size:
113
+ value: 16
114
+ chunk_size:
115
+ value: 50
116
+ compile_mode:
117
+ value: max-autotune
118
+ dataset:
119
+ value: Daankrol/pick-and-place-multi-obj
120
+ dataset_stats:
121
+ value:
122
+ action:
123
+ max:
124
+ - 81.91695404052734
125
+ - 72.79933166503906
126
+ - 100
127
+ - 93.32191467285156
128
+ - -2.732527494430542
129
+ - 42.88956069946289
130
+ mean:
131
+ - 0.47757554785419337
132
+ - -18.034652611064505
133
+ - 15.59805635303217
134
+ - 54.54463619743119
135
+ - -49.21284246567293
136
+ - 7.302610387584375
137
+ min:
138
+ - -75.1649169921875
139
+ - -100
140
+ - -61.050724029541016
141
+ - -6.849315166473389
142
+ - -86.33736419677734
143
+ - 0
144
+ name: action
145
+ q01:
146
+ - -41.29065982641405
147
+ - -100
148
+ - -27.74406375139188
149
+ - 24.550995379247542
150
+ - -61.591176330648565
151
+ - -1.000000013351432e-10
152
+ q99:
153
+ - 34.99315845909429
154
+ - 38.49093164574106
155
+ - 99.99756689137342
156
+ - 82.60630455329574
157
+ - -36.44491567201352
158
+ - 26.6719993779817
159
+ shape:
160
+ - 6
161
+ std:
162
+ - 29.091115680540447
163
+ - 44.16383911983223
164
+ - 44.81775979436524
165
+ - 19.76430732442855
166
+ - 8.799572987673075
167
+ - 10.41276650889744
168
+ type: ACTION
169
+ observation.gripper:
170
+ max:
171
+ - - - 1
172
+ - - - 1
173
+ - - - 1
174
+ mean:
175
+ - - - 0.5900507834223626
176
+ - - - 0.5872656129005113
177
+ - - - 0.5966057530402353
178
+ min:
179
+ - - - 0
180
+ - - - 0
181
+ - - - 0
182
+ name: gripper
183
+ q01:
184
+ - - - 0.07565626654408032
185
+ - - - 0.07064545037186447
186
+ - - - 0.09675786695048419
187
+ q99:
188
+ - - - 0.8841095335313836
189
+ - - - 0.9249016521748583
190
+ - - - 0.9647414252015477
191
+ shape:
192
+ - 3
193
+ - 480
194
+ - 640
195
+ std:
196
+ - - - 0.1831069655022816
197
+ - - - 0.18826039482586307
198
+ - - - 0.18464241941191994
199
+ type: VISUAL
200
+ observation.overhead:
201
+ max:
202
+ - - - 1
203
+ - - - 1
204
+ - - - 1
205
+ mean:
206
+ - - - 0.545862979096059
207
+ - - - 0.5452054301949182
208
+ - - - 0.5477746602359932
209
+ min:
210
+ - - - 0
211
+ - - - 0
212
+ - - - 0
213
+ name: overhead
214
+ q01:
215
+ - - - 0.05494559589188223
216
+ - - - 0.05129975818181184
217
+ - - - 0.07399745843495421
218
+ q99:
219
+ - - - 0.7951740036832365
220
+ - - - 0.8620514435976284
221
+ - - - 0.9287900526200447
222
+ shape:
223
+ - 3
224
+ - 480
225
+ - 640
226
+ std:
227
+ - - - 0.14197105669614393
228
+ - - - 0.1436179498089634
229
+ - - - 0.14500463044962006
230
+ type: VISUAL
231
+ observation.state:
232
+ max:
233
+ - 80.10877990722656
234
+ - 73.74842071533203
235
+ - 98.82192993164062
236
+ - 92.56092071533203
237
+ - -3.136435031890869
238
+ - 42.084434509277344
239
+ mean:
240
+ - 0.4902085707550287
241
+ - -15.791447722340527
242
+ - 18.025905360949306
243
+ - 55.00042447703383
244
+ - -49.21564963228982
245
+ - 9.598709224247822
246
+ min:
247
+ - -71.17327117919922
248
+ - -99.66344451904297
249
+ - -56.68328094482422
250
+ - -6.712270259857178
251
+ - -85.72921752929688
252
+ - 0.26385223865509033
253
+ name: state
254
+ q01:
255
+ - -39.45999045040392
256
+ - -98.81794802740971
257
+ - -23.48391958469011
258
+ - 25.50333895589101
259
+ - -61.34767717791836
260
+ - 0.6040729837518146
261
+ q99:
262
+ - 34.18319318983752
263
+ - 39.30228886401892
264
+ - 98.56118793779315
265
+ - 82.42731621305099
266
+ - -36.652074374338476
267
+ - 26.018496216201445
268
+ shape:
269
+ - 6
270
+ std:
271
+ - 28.56218428477719
272
+ - 44.62872421028026
273
+ - 43.38945126960788
274
+ - 19.454495761579995
275
+ - 8.781436382129517
276
+ - 8.932825748525975
277
+ type: STATE
278
+ devices:
279
+ value: 1
280
+ dtype:
281
+ value: bfloat16
282
+ empty_cameras:
283
+ value: 0
284
+ epochs:
285
+ value: 3
286
+ freeze_vision_encoder:
287
+ value: false
288
+ global_batch:
289
+ value: 16
290
+ gradient_checkpointing:
291
+ value: true
292
+ image_resolution:
293
+ value:
294
+ - 224
295
+ - 224
296
+ max_action_dim:
297
+ value: 32
298
+ max_period:
299
+ value: 4
300
+ max_state_dim:
301
+ value: 32
302
+ min_period:
303
+ value: 0.004
304
+ n_action_steps:
305
+ value: 50
306
+ n_obs_steps:
307
+ value: 1
308
+ normalization_mode:
309
+ value: QUANTILES
310
+ num_inference_steps:
311
+ value: 10
312
+ optimizer_betas:
313
+ value:
314
+ - 0.9
315
+ - 0.95
316
+ optimizer_eps:
317
+ value: 1e-08
318
+ optimizer_grad_clip_norm:
319
+ value: 1
320
+ optimizer_lr:
321
+ value: 2.5e-05
322
+ optimizer_weight_decay:
323
+ value: 0.01
324
+ paligemma_variant:
325
+ value: gemma_2b
326
+ phase:
327
+ value: phase2-snapflow
328
+ pretrained:
329
+ value: lerobot/pi05_base
330
+ scheduler_decay_lr:
331
+ value: 2.5e-06
332
+ scheduler_decay_steps:
333
+ value: null
334
+ scheduler_warmup_steps:
335
+ value: 807
336
+ seed:
337
+ value: 42
338
+ snapflow_alpha:
339
+ value: 0.5
340
+ snapflow_enabled:
341
+ value: true
342
+ snapflow_lambda:
343
+ value: 0.1
344
+ snapflow_num_inference_steps:
345
+ value: 1
346
+ snapflow_steps:
347
+ value: 1
348
+ steps_per_epoch:
349
+ value: 5380
350
+ time_sampling_beta_alpha:
351
+ value: 1.5
352
+ time_sampling_beta_beta:
353
+ value: 1
354
+ time_sampling_offset:
355
+ value: 0.001
356
+ time_sampling_scale:
357
+ value: 0.999
358
+ tokenizer_max_length:
359
+ value: 200
360
+ total_steps:
361
+ value: 16140
362
+ train_expert_only:
363
+ value: true
364
+ train_frames:
365
+ value: 86093
366
+ use_random_input_noise:
367
+ value: true
368
+ val_split:
369
+ value: 0.1
370
+ val_split_seed:
371
+ value: 42
372
+ warmup_frac:
373
+ value: 0.05
374
+ warmup_steps:
375
+ value: 807
phase2_snapflow/wandb/run-20260806_235610-358tcehl/files/output.log ADDED
The diff for this file is too large to render. See raw diff
 
phase2_snapflow/wandb/run-20260806_235610-358tcehl/files/requirements.txt ADDED
@@ -0,0 +1,209 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ av==15.1.0
2
+ multidict==6.7.1
3
+ librt==0.13.0
4
+ etils==1.14.0
5
+ peft==0.19.1
6
+ opencv-python==5.0.0.93
7
+ opencv-python-headless==4.13.0.92
8
+ shtab==1.8.1
9
+ aiohttp==3.14.1
10
+ psutil==7.2.2
11
+ xxhash==3.8.1
12
+ Jinja2==3.1.6
13
+ pyrefly==1.1.1
14
+ h11==0.16.0
15
+ numpy==2.2.6
16
+ draccus==0.10.0
17
+ annotated-doc==0.0.4
18
+ bddl==1.0.1
19
+ tensorboard==2.21.0
20
+ Pygments==2.20.0
21
+ PyTurboJPEG==1.8.3
22
+ openvino==2026.1.0
23
+ PyOpenGL==3.1.10
24
+ filelock==3.31.0
25
+ torch==2.11.0+cu128
26
+ idna==3.18
27
+ pytorch-lightning==2.6.5
28
+ scikit-image==0.26.0
29
+ yarl==1.24.2
30
+ flatbuffers==25.12.19
31
+ pillow==12.3.0
32
+ nvidia-nvtx-cu12==12.8.90
33
+ mpmath==1.3.0
34
+ nbformat==5.10.4
35
+ Werkzeug==3.1.8
36
+ mujoco==3.8.1
37
+ aiosignal==1.4.0
38
+ pydantic==2.13.4
39
+ zipp==4.1.0
40
+ shellingham==1.5.4
41
+ robomimic==0.2.0
42
+ typeshed_client==2.12.0
43
+ annotated-types==0.7.0
44
+ tokenizers==0.22.2
45
+ pandas==2.3.3
46
+ Farama-Notifications==0.0.6
47
+ typer==0.27.0
48
+ tensorboardX==2.6.5
49
+ egl_probe==1.0.2
50
+ fonttools==4.63.0
51
+ pydantic_core==2.46.4
52
+ onnx-ir==0.2.1
53
+ httpx==0.28.1
54
+ mergedeep==1.3.4
55
+ pyarrow==25.0.0
56
+ hf-xet==1.5.2
57
+ pluggy==1.6.0
58
+ tensorboard-data-server==0.7.2
59
+ robosuite==1.4.0
60
+ pyyaml-include==1.4.1
61
+ charset-normalizer==3.4.9
62
+ typing-inspection==0.4.2
63
+ lerobot==0.6.0
64
+ jsonschema-specifications==2025.9.1
65
+ httpcore==1.0.9
66
+ gym-pusht==0.1.6
67
+ onnx==1.22.0
68
+ jupyter_core==5.9.1
69
+ llvmlite==0.48.0
70
+ pathspec==1.1.1
71
+ nvidia-nvjitlink-cu12==12.8.93
72
+ cuda-toolkit==12.8.1
73
+ cffi==2.1.0
74
+ sentencepiece==0.2.1
75
+ pynokhwa==0.1.2
76
+ iniconfig==2.3.0
77
+ nvidia-nccl-cu12==2.28.9
78
+ urllib3==2.7.0
79
+ jsonargparse==4.49.0
80
+ jupytext==1.19.4
81
+ referencing==0.37.0
82
+ grpcio==1.82.1
83
+ Markdown==3.10.2
84
+ torchvision==0.26.0+cu128
85
+ frozenlist==1.8.0
86
+ pygame==2.6.1
87
+ typing-inspect==0.9.0
88
+ h5py==3.16.0
89
+ typing_extensions==4.16.0
90
+ imageio-ffmpeg==0.6.0
91
+ packaging==25.0
92
+ rich==15.0.0
93
+ ImageIO==2.37.3
94
+ protobuf==7.35.1
95
+ mypy==2.3.0
96
+ ast_serialize==0.6.0
97
+ scipy==1.18.0
98
+ shapely==2.1.2
99
+ mdurl==0.1.2
100
+ multiprocess==0.70.19
101
+ num2words==0.5.14
102
+ hf_libero==0.1.4
103
+ mypy_extensions==1.1.0
104
+ cmake==4.1.3
105
+ openvino-telemetry==2025.2.0
106
+ nvidia-cuda-cupti-cu12==12.8.90
107
+ toml==0.10.2
108
+ thop==0.1.1-2209072238
109
+ huggingface_hub==1.24.0
110
+ numba==0.66.0
111
+ click==8.4.2
112
+ python-dateutil==2.9.0.post0
113
+ safetensors==0.8.0
114
+ absl-py==2.5.0
115
+ markdown-it-py==4.2.0
116
+ nvidia-cusolver-cu12==11.7.3.90
117
+ nvidia-cudnn-cu12==9.19.0.56
118
+ setuptools==80.10.2
119
+ cuda-bindings==12.9.7
120
+ aiohappyeyeballs==2.7.1
121
+ contourpy==1.3.3
122
+ glfw==2.10.1
123
+ pytest==9.1.1
124
+ tqdm==4.69.0
125
+ networkx==3.4.2
126
+ dill==0.4.1
127
+ MarkupSafe==3.0.3
128
+ rpds-py==2026.6.3
129
+ nvidia-cusparselt-cu12==0.7.1
130
+ PyYAML==6.0.3
131
+ importlib_resources==7.1.0
132
+ openvino-tokenizers==2026.1.0.0
133
+ pycparser==3.0
134
+ kiwisolver==1.5.0
135
+ cloudpickle==3.1.2
136
+ lightning==2.6.5
137
+ platformdirs==4.10.1
138
+ docstring_parser==0.18.0
139
+ regex==2026.7.19
140
+ docopt==0.6.2
141
+ tifffile==2026.7.14
142
+ cycler==0.12.1
143
+ nvidia-cuda-runtime-cu12==12.8.90
144
+ nvidia-nvshmem-cu12==3.4.5
145
+ ml_dtypes==0.5.4
146
+ transformers==5.5.4
147
+ nvidia-cufft-cu12==11.3.3.83
148
+ einops==0.8.2
149
+ jsonlines==4.0.0
150
+ wandb==0.28.1
151
+ onnxscript==0.7.1
152
+ nvidia-cuda-nvrtc-cu12==12.8.93
153
+ nvidia-cusparse-cu12==12.5.8.93
154
+ nvidia-cufile-cu12==1.13.1.3
155
+ nvidia-curand-cu12==10.3.9.90
156
+ jsonschema==4.26.0
157
+ lazy-loader==0.5
158
+ gymnasium==1.3.0
159
+ onnxruntime==1.27.0
160
+ torchcodec==0.11.1
161
+ six==1.17.0
162
+ physicalai==0.1.2.dev37+g3e163cfeb
163
+ sentry-sdk==2.66.0
164
+ omegaconf==2.3.1
165
+ attrs==26.1.0
166
+ future==1.0.0
167
+ tzdata==2026.3
168
+ antlr4-python3-runtime==4.9.3
169
+ pyparsing==3.3.2
170
+ fsspec==2026.2.0
171
+ torchmetrics==1.9.0
172
+ anyio==4.14.2
173
+ termcolor==3.3.0
174
+ accelerate==1.14.0
175
+ traitlets==5.15.1
176
+ requests==2.34.2
177
+ physicalai-train==0.1.0
178
+ triton==3.6.0
179
+ easydict==1.13
180
+ ruff==0.15.22
181
+ certifi==2026.6.17
182
+ matplotlib==3.11.1
183
+ hf_egl_probe==1.0.2
184
+ diffusers==0.39.0
185
+ lightning-utilities==0.15.3
186
+ cuda-pathfinder==1.5.6
187
+ sympy==1.14.0
188
+ loguru==0.7.3
189
+ datasets==4.8.5
190
+ nvidia-cublas-cu12==12.8.4.1
191
+ importlib_metadata==9.0.0
192
+ fastjsonschema==2.21.2
193
+ pymunk==6.11.1
194
+ pytz==2026.2
195
+ hydra-core==1.3.4
196
+ propcache==0.5.2
197
+ mdit-py-plugins==0.6.1
198
+ tomli==2.4.0
199
+ more-itertools==10.8.0
200
+ autocommand==2.2.2
201
+ platformdirs==4.4.0
202
+ wheel==0.46.3
203
+ zipp==3.23.0
204
+ backports.tarfile==1.2.0
205
+ jaraco.functools==4.4.0
206
+ packaging==26.0
207
+ importlib_metadata==8.7.1
208
+ jaraco.text==4.0.0
209
+ jaraco.context==6.1.0
phase2_snapflow/wandb/run-20260806_235610-358tcehl/files/wandb-metadata.json ADDED
@@ -0,0 +1,71 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-136-generic-x86_64-with-glibc2.39",
3
+ "python": "CPython 3.12.3",
4
+ "startedAt": "2026-08-06T23:56:10.250950Z",
5
+ "args": [
6
+ "--wandb",
7
+ "--augment",
8
+ "--phase1-epochs",
9
+ "7",
10
+ "--dataset",
11
+ "Daankrol/pick-and-place-multi-obj",
12
+ "--output-dir",
13
+ "/mnt/data/experiments/snapflow-multi-obj"
14
+ ],
15
+ "program": "/home/devuser/daankrol/physical-ai-studio/library/scripts/train_snapflow_so101.py",
16
+ "codePath": "library/scripts/train_snapflow_so101.py",
17
+ "codePathLocal": "library/scripts/train_snapflow_so101.py",
18
+ "git": {
19
+ "remote": "https://github.com/open-edge-platform/physical-ai-studio.git",
20
+ "commit": "f3a674359c427756b3abf7fe50766f03bc3da2d4"
21
+ },
22
+ "email": "daankrol@hotmail.com",
23
+ "root": "/mnt/data/experiments/snapflow-multi-obj/phase2_snapflow",
24
+ "host": "ikvensx013",
25
+ "executable": "/home/devuser/daankrol/physical-ai-studio/library/.venv/bin/python",
26
+ "cpu_count": 64,
27
+ "cpu_count_logical": 128,
28
+ "gpu": "NVIDIA H100 PCIe",
29
+ "gpu_count": 4,
30
+ "disk": {
31
+ "/": {
32
+ "total": "1887976378368",
33
+ "used": "1660689944576"
34
+ }
35
+ },
36
+ "memory": {
37
+ "total": "1352432504832"
38
+ },
39
+ "gpu_nvidia": [
40
+ {
41
+ "name": "NVIDIA H100 PCIe",
42
+ "memoryTotal": "85520809984",
43
+ "cudaCores": 14592,
44
+ "architecture": "Hopper",
45
+ "uuid": "GPU-99931a32-3903-4dd2-57aa-3f5a87cceab6"
46
+ },
47
+ {
48
+ "name": "NVIDIA H100 PCIe",
49
+ "memoryTotal": "85520809984",
50
+ "cudaCores": 14592,
51
+ "architecture": "Hopper",
52
+ "uuid": "GPU-48a80548-2790-2ee9-d239-752ba918214e"
53
+ },
54
+ {
55
+ "name": "NVIDIA H100 PCIe",
56
+ "memoryTotal": "85520809984",
57
+ "cudaCores": 14592,
58
+ "architecture": "Hopper",
59
+ "uuid": "GPU-dcc57cb5-b02f-ab7f-751d-4d55df91356d"
60
+ },
61
+ {
62
+ "name": "NVIDIA H100 PCIe",
63
+ "memoryTotal": "85520809984",
64
+ "cudaCores": 14592,
65
+ "architecture": "Hopper",
66
+ "uuid": "GPU-73153eba-ee9b-1bd6-c1cc-2f14527d0497"
67
+ }
68
+ ],
69
+ "cudaVersion": "13.2",
70
+ "writerId": "0nwnfyvby6u9m7fidg3frxdoi5x5535q"
71
+ }
phase2_snapflow/wandb/run-20260806_235610-358tcehl/files/wandb-summary.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"_timestamp":1.786072287615868e+09,"_wandb":{"runtime":11723},"train/loss":0.0019395967246964574,"_step":646,"_runtime":11723,"val/loss":0.09250237792730331,"trainer/global_step":16139,"epoch":2,"lr-AdamW":2.5003582453977483e-06}
phase2_snapflow/wandb/run-20260806_235610-358tcehl/logs/debug-core.log ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {"time":"2026-08-06T13:49:58.185057882Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpj5sbudda/port-4075244.txt","pid":4075244,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2026-08-06T13:49:58.185731683Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":4075244}
3
+ {"time":"2026-08-06T13:49:58.185719986Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-4075244-4080653-190553848/socket","Net":"unix"}}
4
+ {"time":"2026-08-06T13:49:58.357289206Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
5
+ {"time":"2026-08-06T13:49:58.674606405Z","level":"INFO","msg":"handleInformInit: received","streamId":"uwqkxwz8","id":"1(@)"}
6
+ {"time":"2026-08-06T13:49:58.795058655Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"uwqkxwz8","id":"1(@)"}
7
+ {"time":"2026-08-06T13:50:04.650124815Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"wf0imehqa7sc"}
8
+ {"time":"2026-08-06T23:54:26.914950247Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"wf0imehqa7sc"}
9
+ {"time":"2026-08-06T23:54:28.756267886Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"wf0imehqa7sc"}
10
+ {"time":"2026-08-06T23:54:28.758376429Z","level":"INFO","msg":"handleInformFinish: finish message received","streamId":"uwqkxwz8","id":"1(@)"}
11
+ {"time":"2026-08-06T23:54:28.75964422Z","level":"INFO","msg":"handleInformFinish: stream closed","streamId":"uwqkxwz8","id":"1(@)"}
12
+ {"time":"2026-08-06T23:56:10.254631266Z","level":"INFO","msg":"handleInformInit: received","streamId":"358tcehl","id":"1(@)"}
13
+ {"time":"2026-08-06T23:56:10.368713975Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"358tcehl","id":"1(@)"}
14
+ {"time":"2026-08-06T23:56:16.483632326Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"ku5fxa4z53xf"}
15
+ {"time":"2026-08-07T03:11:34.299282741Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"ku5fxa4z53xf"}
16
+ {"time":"2026-08-07T03:11:36.472611773Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"ku5fxa4z53xf"}
17
+ {"time":"2026-08-07T03:11:36.474246789Z","level":"INFO","msg":"handleInformFinish: finish message received","streamId":"358tcehl","id":"1(@)"}
18
+ {"time":"2026-08-07T03:11:36.475743729Z","level":"INFO","msg":"handleInformFinish: stream closed","streamId":"358tcehl","id":"1(@)"}
19
+ {"time":"2026-08-07T03:11:36.475965645Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
20
+ {"time":"2026-08-07T03:11:36.475986263Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
21
+ {"time":"2026-08-07T03:11:36.475997356Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
22
+ {"time":"2026-08-07T03:11:36.47605653Z","level":"INFO","msg":"server: is shutting down"}
23
+ {"time":"2026-08-07T03:11:36.476066882Z","level":"INFO","msg":"processOutgoingData: finished","id":"1(@)"}
24
+ {"time":"2026-08-07T03:11:36.476078148Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
25
+ {"time":"2026-08-07T03:11:36.47611314Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
26
+ {"time":"2026-08-07T03:11:36.476282677Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-4075244-4080653-190553848/socket","Net":"unix"}}
27
+ {"time":"2026-08-07T03:11:36.476359942Z","level":"INFO","msg":"server: all connections closed"}
phase2_snapflow/wandb/run-20260806_235610-358tcehl/logs/debug-internal.log ADDED
The diff for this file is too large to render. See raw diff
 
phase2_snapflow/wandb/run-20260806_235610-358tcehl/logs/debug.log ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-08-06 23:56:10,252 INFO MainThread:4075244 [wandb_init.py:setup_run_log_directory():729] Logging user logs to /mnt/data/experiments/snapflow-multi-obj/phase2_snapflow/wandb/run-20260806_235610-358tcehl/logs/debug.log
2
+ 2026-08-06 23:56:10,252 INFO MainThread:4075244 [wandb_init.py:setup_run_log_directory():730] Logging internal logs to /mnt/data/experiments/snapflow-multi-obj/phase2_snapflow/wandb/run-20260806_235610-358tcehl/logs/debug-internal.log
3
+ 2026-08-06 23:56:10,252 INFO MainThread:4075244 [wandb_init.py:init():772] calling init triggers
4
+ 2026-08-06 23:56:10,252 INFO MainThread:4075244 [wandb_init.py:init():777] wandb.init called with sweep_config: {}
5
+ config: {'_wandb': {}}
6
+ 2026-08-06 23:56:10,253 INFO MainThread:4075244 [wandb_init.py:init():820] starting backend
7
+ 2026-08-06 23:56:10,253 INFO MainThread:4075244 [wandb_init.py:init():835] sending inform_init request
8
+ 2026-08-06 23:56:10,369 INFO MainThread:4075244 [wandb_init.py:init():840] backend started and connected
9
+ 2026-08-06 23:56:10,378 INFO MainThread:4075244 [wandb_init.py:init():910] updated telemetry
10
+ 2026-08-06 23:56:10,388 INFO MainThread:4075244 [wandb_init.py:init():933] communicating run to backend with 90.0 second timeout
11
+ 2026-08-06 23:56:11,071 INFO MainThread:4075244 [wandb_init.py:init():978] starting run threads in backend
12
+ 2026-08-06 23:56:11,244 INFO MainThread:4075244 [wandb_run.py:_console_start():2621] atexit reg
13
+ 2026-08-06 23:56:11,244 INFO MainThread:4075244 [wandb_run.py:_redirect():2471] redirect: wrap_raw
14
+ 2026-08-06 23:56:11,244 INFO MainThread:4075244 [wandb_run.py:_redirect():2540] Wrapping output streams.
15
+ 2026-08-06 23:56:11,244 INFO MainThread:4075244 [wandb_run.py:_redirect():2563] Redirects installed.
16
+ 2026-08-06 23:56:11,245 INFO MainThread:4075244 [wandb_init.py:init():1016] run started, returning control to user process
17
+ 2026-08-06 23:56:11,246 INFO MainThread:4075244 [wandb_run.py:_config_callback():1346] config_cb None None {'phase': 'phase2-snapflow', 'dataset': 'Daankrol/pick-and-place-multi-obj', 'pretrained': 'lerobot/pi05_base', 'seed': 42, 'val_split': 0.1, 'val_split_seed': 42, 'augment': True, 'augment_n_subset': 3, 'batch_size': 16, 'devices': 1, 'accumulate_grad_batches': 1, 'epochs': 3, 'train_frames': 86093, 'global_batch': 16, 'steps_per_epoch': 5380, 'total_steps': 16140, 'warmup_steps': 807, 'warmup_frac': 0.05, 'snapflow_alpha': 0.5, 'snapflow_lambda': 0.1, 'snapflow_steps': 1}
18
+ 2026-08-06 23:56:16,547 INFO MainThread:4075244 [wandb_run.py:_config_callback():1346] config_cb None None {'paligemma_variant': 'gemma_2b', 'action_expert_variant': 'gemma_300m', 'dtype': 'bfloat16', 'n_obs_steps': 1, 'chunk_size': 50, 'n_action_steps': 50, 'max_state_dim': 32, 'max_action_dim': 32, 'num_inference_steps': 10, 'time_sampling_beta_alpha': 1.5, 'time_sampling_beta_beta': 1.0, 'time_sampling_scale': 0.999, 'time_sampling_offset': 0.001, 'min_period': 0.004, 'max_period': 4.0, 'use_random_input_noise': True, 'snapflow_enabled': True, 'snapflow_alpha': 0.5, 'snapflow_lambda': 0.1, 'snapflow_num_inference_steps': 1, 'image_resolution': [224, 224], 'empty_cameras': 0, 'tokenizer_max_length': 200, 'gradient_checkpointing': True, 'compile_mode': 'max-autotune', 'freeze_vision_encoder': False, 'train_expert_only': True, 'normalization_mode': 'QUANTILES', 'optimizer_lr': 2.5e-05, 'optimizer_betas': [0.9, 0.95], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 0.01, 'optimizer_grad_clip_norm': 1.0, 'scheduler_warmup_steps': 807, 'scheduler_decay_steps': None, 'scheduler_decay_lr': 2.5e-06, 'dataset_stats': {'observation.state': {'mean': [0.4902085707550287, -15.791447722340527, 18.025905360949306, 55.00042447703383, -49.21564963228982, 9.598709224247822], 'std': [28.56218428477719, 44.62872421028026, 43.38945126960788, 19.454495761579995, 8.781436382129517, 8.932825748525975], 'min': [-71.17327117919922, -99.66344451904297, -56.68328094482422, -6.712270259857178, -85.72921752929688, 0.26385223865509033], 'max': [80.10877990722656, 73.74842071533203, 98.82192993164062, 92.56092071533203, -3.136435031890869, 42.084434509277344], 'q01': [-39.45999045040392, -98.81794802740971, -23.48391958469011, 25.50333895589101, -61.34767717791836, 0.6040729837518146], 'q99': [34.18319318983752, 39.30228886401892, 98.56118793779315, 82.42731621305099, -36.652074374338476, 26.018496216201445], 'type': 'STATE', 'name': 'state', 'shape': [6]}, 'observation.gripper': {'mean': [[[0.5900507834223626]], [[0.5872656129005113]], [[0.5966057530402353]]], 'std': [[[0.1831069655022816]], [[0.18826039482586307]], [[0.18464241941191994]]], 'min': [[[0.0]], [[0.0]], [[0.0]]], 'max': [[[1.0]], [[1.0]], [[1.0]]], 'q01': [[[0.07565626654408032]], [[0.07064545037186447]], [[0.09675786695048419]]], 'q99': [[[0.8841095335313836]], [[0.9249016521748583]], [[0.9647414252015477]]], 'type': 'VISUAL', 'name': 'gripper', 'shape': [3, 480, 640]}, 'observation.overhead': {'mean': [[[0.545862979096059]], [[0.5452054301949182]], [[0.5477746602359932]]], 'std': [[[0.14197105669614393]], [[0.1436179498089634]], [[0.14500463044962006]]], 'min': [[[0.0]], [[0.0]], [[0.0]]], 'max': [[[1.0]], [[1.0]], [[1.0]]], 'q01': [[[0.05494559589188223]], [[0.05129975818181184]], [[0.07399745843495421]]], 'q99': [[[0.7951740036832365]], [[0.8620514435976284]], [[0.9287900526200447]]], 'type': 'VISUAL', 'name': 'overhead', 'shape': [3, 480, 640]}, 'action': {'mean': [0.47757554785419337, -18.034652611064505, 15.59805635303217, 54.54463619743119, -49.21284246567293, 7.302610387584375], 'std': [29.091115680540447, 44.16383911983223, 44.81775979436524, 19.76430732442855, 8.799572987673075, 10.41276650889744], 'min': [-75.1649169921875, -100.0, -61.050724029541016, -6.849315166473389, -86.33736419677734, 0.0], 'max': [81.91695404052734, 72.79933166503906, 100.0, 93.32191467285156, -2.732527494430542, 42.88956069946289], 'q01': [-41.29065982641405, -100.0, -27.74406375139188, 24.550995379247542, -61.591176330648565, -1.000000013351432e-10], 'q99': [34.99315845909429, 38.49093164574106, 99.99756689137342, 82.60630455329574, -36.44491567201352, 26.6719993779817], 'type': 'ACTION', 'name': 'action', 'shape': [6]}}}
19
+ 2026-08-07 03:11:34,297 INFO MainThread:4075244 [wandb_run.py:_finish():2383] finishing run daankrol/physicalai_snapflow/358tcehl
20
+ 2026-08-07 03:11:34,298 INFO MainThread:4075244 [wandb_run.py:_atexit_cleanup():2588] got exitcode: 0
21
+ 2026-08-07 03:11:34,298 INFO MainThread:4075244 [wandb_run.py:_restore():2570] restore
22
+ 2026-08-07 03:11:34,298 INFO MainThread:4075244 [wandb_run.py:_restore():2576] restore done
23
+ 2026-08-07 03:11:36,473 INFO MainThread:4075244 [wandb_run.py:_footer_sync_info():3993] logging synced files
phase2_snapflow/wandb/run-20260806_235610-358tcehl/run-358tcehl.wandb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:708ab9060d307ad04334b051844c8e4719b1fd441d85e421fab9f7f839dd85ce
3
+ size 15126959
phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/files/output.log ADDED
The diff for this file is too large to render. See raw diff
 
phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/files/requirements.txt ADDED
@@ -0,0 +1,158 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ av==15.1.0
2
+ multidict==6.7.1
3
+ librt==0.13.0
4
+ peft==0.19.1
5
+ opencv-python==5.0.0.93
6
+ opencv-python-headless==4.13.0.92
7
+ shtab==1.8.1
8
+ aiohttp==3.14.1
9
+ psutil==7.2.2
10
+ xxhash==3.8.1
11
+ Jinja2==3.1.6
12
+ pyrefly==1.1.1
13
+ h11==0.16.0
14
+ numpy==2.2.6
15
+ draccus==0.10.0
16
+ annotated-doc==0.0.4
17
+ Pygments==2.20.0
18
+ PyTurboJPEG==1.8.3
19
+ openvino==2026.1.0
20
+ filelock==3.31.0
21
+ torch==2.11.0+cu128
22
+ idna==3.18
23
+ pytorch-lightning==2.6.5
24
+ scikit-image==0.26.0
25
+ yarl==1.24.2
26
+ flatbuffers==25.12.19
27
+ pillow==12.3.0
28
+ nvidia-nvtx-cu12==12.8.90
29
+ mpmath==1.3.0
30
+ aiosignal==1.4.0
31
+ pydantic==2.13.4
32
+ zipp==4.1.0
33
+ shellingham==1.5.4
34
+ typeshed_client==2.12.0
35
+ annotated-types==0.7.0
36
+ tokenizers==0.22.2
37
+ pandas==2.3.3
38
+ Farama-Notifications==0.0.6
39
+ typer==0.27.0
40
+ pydantic_core==2.46.4
41
+ onnx-ir==0.2.1
42
+ httpx==0.28.1
43
+ mergedeep==1.3.4
44
+ pyarrow==25.0.0
45
+ hf-xet==1.5.2
46
+ pluggy==1.6.0
47
+ pytest-mock==3.15.1
48
+ pyyaml-include==1.4.1
49
+ charset-normalizer==3.4.9
50
+ typing-inspection==0.4.2
51
+ lerobot==0.6.0
52
+ httpcore==1.0.9
53
+ gym-pusht==0.1.6
54
+ onnx==1.22.0
55
+ pathspec==1.1.1
56
+ nvidia-nvjitlink-cu12==12.8.93
57
+ cuda-toolkit==12.8.1
58
+ cffi==2.1.0
59
+ sentencepiece==0.2.1
60
+ pynokhwa==0.1.2
61
+ iniconfig==2.3.0
62
+ nvidia-nccl-cu12==2.28.9
63
+ urllib3==2.7.0
64
+ jsonargparse==4.49.0
65
+ torchvision==0.26.0+cu128
66
+ frozenlist==1.8.0
67
+ pygame==2.6.1
68
+ typing-inspect==0.9.0
69
+ typing_extensions==4.16.0
70
+ packaging==25.0
71
+ rich==15.0.0
72
+ ImageIO==2.37.3
73
+ protobuf==7.35.1
74
+ mypy==2.3.0
75
+ ast_serialize==0.6.0
76
+ scipy==1.18.0
77
+ shapely==2.1.2
78
+ mdurl==0.1.2
79
+ multiprocess==0.70.19
80
+ num2words==0.5.14
81
+ mypy_extensions==1.1.0
82
+ cmake==4.1.3
83
+ openvino-telemetry==2025.2.0
84
+ nvidia-cuda-cupti-cu12==12.8.90
85
+ toml==0.10.2
86
+ huggingface_hub==1.24.0
87
+ click==8.4.2
88
+ python-dateutil==2.9.0.post0
89
+ safetensors==0.8.0
90
+ pytest-examples==0.0.18
91
+ markdown-it-py==4.2.0
92
+ nvidia-cusolver-cu12==11.7.3.90
93
+ nvidia-cudnn-cu12==9.19.0.56
94
+ setuptools==80.10.2
95
+ cuda-bindings==12.9.7
96
+ aiohappyeyeballs==2.7.1
97
+ pytest==9.1.1
98
+ tqdm==4.69.0
99
+ networkx==3.4.2
100
+ dill==0.4.1
101
+ MarkupSafe==3.0.3
102
+ nvidia-cusparselt-cu12==0.7.1
103
+ PyYAML==6.0.3
104
+ importlib_resources==7.1.0
105
+ openvino-tokenizers==2026.1.0.0
106
+ pycparser==3.0
107
+ sentry-sdk==2.66.1
108
+ cloudpickle==3.1.2
109
+ lightning==2.6.5
110
+ platformdirs==4.10.1
111
+ docstring_parser==0.18.0
112
+ regex==2026.7.19
113
+ docopt==0.6.2
114
+ tifffile==2026.7.14
115
+ nvidia-cuda-runtime-cu12==12.8.90
116
+ nvidia-nvshmem-cu12==3.4.5
117
+ ml_dtypes==0.5.4
118
+ transformers==5.5.4
119
+ nvidia-cufft-cu12==11.3.3.83
120
+ einops==0.8.2
121
+ jsonlines==4.0.0
122
+ wandb==0.28.1
123
+ onnxscript==0.7.1
124
+ nvidia-cuda-nvrtc-cu12==12.8.93
125
+ nvidia-cusparse-cu12==12.5.8.93
126
+ nvidia-cufile-cu12==1.13.1.3
127
+ nvidia-curand-cu12==10.3.9.90
128
+ lazy-loader==0.5
129
+ gymnasium==1.3.0
130
+ onnxruntime==1.27.0
131
+ torchcodec==0.11.1
132
+ six==1.17.0
133
+ physicalai==0.1.2.dev37+g3e163cfeb
134
+ attrs==26.1.0
135
+ tzdata==2026.3
136
+ fsspec==2026.2.0
137
+ torchmetrics==1.9.0
138
+ anyio==4.14.2
139
+ termcolor==3.3.0
140
+ accelerate==1.14.0
141
+ requests==2.34.2
142
+ black==26.5.1
143
+ physicalai-train==0.1.0
144
+ triton==3.6.0
145
+ ruff==0.15.22
146
+ certifi==2026.6.17
147
+ pytokens==0.4.1
148
+ diffusers==0.39.0
149
+ lightning-utilities==0.15.3
150
+ cuda-pathfinder==1.5.6
151
+ sympy==1.14.0
152
+ loguru==0.7.3
153
+ datasets==4.8.5
154
+ nvidia-cublas-cu12==12.8.4.1
155
+ importlib_metadata==9.0.0
156
+ pymunk==6.11.1
157
+ pytz==2026.2
158
+ propcache==0.5.2
phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/files/wandb-metadata.json ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "os": "Linux-6.8.0-136-generic-x86_64-with-glibc2.39",
3
+ "python": "CPython 3.12.3",
4
+ "startedAt": "2026-08-07T09:38:05.626206Z",
5
+ "args": [
6
+ "--wandb",
7
+ "--augment",
8
+ "--dataset",
9
+ "Daankrol/pick-and-place-multi-obj",
10
+ "--output-dir",
11
+ "/mnt/data/experiments/snapflow-multi-obj",
12
+ "--skip-phase1",
13
+ "--phase1-ckpt",
14
+ "/mnt/data/experiments/snapflow-multi-obj/phase1/phase1-epoch005.ckpt"
15
+ ],
16
+ "program": "/home/devuser/daankrol/physical-ai-studio/library/scripts/train_snapflow_so101.py",
17
+ "codePath": "library/scripts/train_snapflow_so101.py",
18
+ "codePathLocal": "library/scripts/train_snapflow_so101.py",
19
+ "git": {
20
+ "remote": "https://github.com/open-edge-platform/physical-ai-studio.git",
21
+ "commit": "f3a674359c427756b3abf7fe50766f03bc3da2d4"
22
+ },
23
+ "email": "daankrol@hotmail.com",
24
+ "root": "/mnt/data/experiments/snapflow-multi-obj/phase2_snapflow",
25
+ "host": "ikvensx013",
26
+ "executable": "/home/devuser/daankrol/physical-ai-studio/library/.venv/bin/python",
27
+ "cpu_count": 64,
28
+ "cpu_count_logical": 128,
29
+ "gpu": "NVIDIA H100 PCIe",
30
+ "gpu_count": 4,
31
+ "disk": {
32
+ "/": {
33
+ "total": "1887976378368",
34
+ "used": "1659492585472"
35
+ }
36
+ },
37
+ "memory": {
38
+ "total": "1352432504832"
39
+ },
40
+ "gpu_nvidia": [
41
+ {
42
+ "name": "NVIDIA H100 PCIe",
43
+ "memoryTotal": "85520809984",
44
+ "cudaCores": 14592,
45
+ "architecture": "Hopper",
46
+ "uuid": "GPU-99931a32-3903-4dd2-57aa-3f5a87cceab6"
47
+ },
48
+ {
49
+ "name": "NVIDIA H100 PCIe",
50
+ "memoryTotal": "85520809984",
51
+ "cudaCores": 14592,
52
+ "architecture": "Hopper",
53
+ "uuid": "GPU-48a80548-2790-2ee9-d239-752ba918214e"
54
+ },
55
+ {
56
+ "name": "NVIDIA H100 PCIe",
57
+ "memoryTotal": "85520809984",
58
+ "cudaCores": 14592,
59
+ "architecture": "Hopper",
60
+ "uuid": "GPU-dcc57cb5-b02f-ab7f-751d-4d55df91356d"
61
+ },
62
+ {
63
+ "name": "NVIDIA H100 PCIe",
64
+ "memoryTotal": "85520809984",
65
+ "cudaCores": 14592,
66
+ "architecture": "Hopper",
67
+ "uuid": "GPU-73153eba-ee9b-1bd6-c1cc-2f14527d0497"
68
+ }
69
+ ],
70
+ "cudaVersion": "13.2",
71
+ "writerId": "r3o95qw7g6pfp1eo27kmd5elgs1q5j2w"
72
+ }
phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/logs/debug-core.log ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {"time":"2026-08-07T09:38:05.166398817Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpbb1t2hw9/port-1916126.txt","pid":1916126,"detached":false,"idle-timeout":600000000000,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
2
+ {"time":"2026-08-07T09:38:05.167312858Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":1916126}
3
+ {"time":"2026-08-07T09:38:05.167278848Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-1916126-1916825-636256715/socket","Net":"unix"}}
4
+ {"time":"2026-08-07T09:38:05.333917539Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
5
+ {"time":"2026-08-07T09:38:05.629345554Z","level":"INFO","msg":"handleInformInit: received","streamId":"7i8cqirl","id":"1(@)"}
6
+ {"time":"2026-08-07T09:38:05.74496059Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"7i8cqirl","id":"1(@)"}
7
+ {"time":"2026-08-07T09:38:11.735225788Z","level":"INFO","msg":"connection: cancelling request","id":"1(@)","requestId":"l7clki87x5l3"}
phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/logs/debug-internal.log ADDED
The diff for this file is too large to render. See raw diff
 
phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/logs/debug.log ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_setup.py:_flush():81] Current SDK version is 0.28.1
2
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_setup.py:_flush():81] Configure stats pid to 1916126
3
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_setup.py:_flush():81] Loading settings from environment variables
4
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:setup_run_log_directory():729] Logging user logs to /mnt/data/experiments/snapflow-multi-obj/phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/logs/debug.log
5
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:setup_run_log_directory():730] Logging internal logs to /mnt/data/experiments/snapflow-multi-obj/phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/logs/debug-internal.log
6
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:init():772] calling init triggers
7
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:init():777] wandb.init called with sweep_config: {}
8
+ config: {'_wandb': {}}
9
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:init():820] starting backend
10
+ 2026-08-07 09:38:05,627 INFO MainThread:1916126 [wandb_init.py:init():835] sending inform_init request
11
+ 2026-08-07 09:38:05,745 INFO MainThread:1916126 [wandb_init.py:init():840] backend started and connected
12
+ 2026-08-07 09:38:05,747 INFO MainThread:1916126 [wandb_init.py:init():910] updated telemetry
13
+ 2026-08-07 09:38:05,755 INFO MainThread:1916126 [wandb_init.py:init():933] communicating run to backend with 90.0 second timeout
14
+ 2026-08-07 09:38:06,580 INFO MainThread:1916126 [wandb_init.py:init():978] starting run threads in backend
15
+ 2026-08-07 09:38:06,731 INFO MainThread:1916126 [wandb_run.py:_console_start():2621] atexit reg
16
+ 2026-08-07 09:38:06,731 INFO MainThread:1916126 [wandb_run.py:_redirect():2471] redirect: wrap_raw
17
+ 2026-08-07 09:38:06,731 INFO MainThread:1916126 [wandb_run.py:_redirect():2540] Wrapping output streams.
18
+ 2026-08-07 09:38:06,731 INFO MainThread:1916126 [wandb_run.py:_redirect():2563] Redirects installed.
19
+ 2026-08-07 09:38:06,738 INFO MainThread:1916126 [wandb_init.py:init():1016] run started, returning control to user process
20
+ 2026-08-07 09:38:06,739 INFO MainThread:1916126 [wandb_run.py:_config_callback():1346] config_cb None None {'phase': 'phase2-snapflow', 'dataset': 'Daankrol/pick-and-place-multi-obj', 'pretrained': 'lerobot/pi05_base', 'seed': 42, 'val_split': 0.1, 'val_split_seed': 42, 'augment': True, 'augment_n_subset': 3, 'batch_size': 16, 'devices': 1, 'accumulate_grad_batches': 1, 'epochs': 3, 'train_frames': 86093, 'global_batch': 16, 'steps_per_epoch': 5380, 'total_steps': 16140, 'warmup_steps': 807, 'warmup_frac': 0.05, 'snapflow_alpha': 0.5, 'snapflow_lambda': 0.1, 'snapflow_steps': 1}
21
+ 2026-08-07 09:38:09,271 INFO MainThread:1916126 [wandb_run.py:_config_callback():1346] config_cb None None {'paligemma_variant': 'gemma_2b', 'action_expert_variant': 'gemma_300m', 'dtype': 'bfloat16', 'n_obs_steps': 1, 'chunk_size': 50, 'n_action_steps': 50, 'max_state_dim': 32, 'max_action_dim': 32, 'num_inference_steps': 10, 'time_sampling_beta_alpha': 1.5, 'time_sampling_beta_beta': 1.0, 'time_sampling_scale': 0.999, 'time_sampling_offset': 0.001, 'min_period': 0.004, 'max_period': 4.0, 'use_random_input_noise': True, 'snapflow_enabled': True, 'snapflow_alpha': 0.5, 'snapflow_lambda': 0.1, 'snapflow_num_inference_steps': 1, 'image_resolution': [224, 224], 'empty_cameras': 0, 'tokenizer_max_length': 200, 'gradient_checkpointing': True, 'compile_mode': 'max-autotune', 'freeze_vision_encoder': False, 'train_expert_only': True, 'normalization_mode': 'QUANTILES', 'optimizer_lr': 2.5e-05, 'optimizer_betas': [0.9, 0.95], 'optimizer_eps': 1e-08, 'optimizer_weight_decay': 0.01, 'optimizer_grad_clip_norm': 1.0, 'scheduler_warmup_steps': 807, 'scheduler_decay_steps': None, 'scheduler_decay_lr': 2.5e-06, 'dataset_stats': {'observation.state': {'mean': [0.4902085707550287, -15.791447722340527, 18.025905360949306, 55.00042447703383, -49.21564963228982, 9.598709224247822], 'std': [28.56218428477719, 44.62872421028026, 43.38945126960788, 19.454495761579995, 8.781436382129517, 8.932825748525975], 'min': [-71.17327117919922, -99.66344451904297, -56.68328094482422, -6.712270259857178, -85.72921752929688, 0.26385223865509033], 'max': [80.10877990722656, 73.74842071533203, 98.82192993164062, 92.56092071533203, -3.136435031890869, 42.084434509277344], 'q01': [-39.45999045040392, -98.81794802740971, -23.48391958469011, 25.50333895589101, -61.34767717791836, 0.6040729837518146], 'q99': [34.18319318983752, 39.30228886401892, 98.56118793779315, 82.42731621305099, -36.652074374338476, 26.018496216201445], 'type': 'STATE', 'name': 'state', 'shape': [6]}, 'observation.gripper': {'mean': [[[0.5900507834223626]], [[0.5872656129005113]], [[0.5966057530402353]]], 'std': [[[0.1831069655022816]], [[0.18826039482586307]], [[0.18464241941191994]]], 'min': [[[0.0]], [[0.0]], [[0.0]]], 'max': [[[1.0]], [[1.0]], [[1.0]]], 'q01': [[[0.07565626654408032]], [[0.07064545037186447]], [[0.09675786695048419]]], 'q99': [[[0.8841095335313836]], [[0.9249016521748583]], [[0.9647414252015477]]], 'type': 'VISUAL', 'name': 'gripper', 'shape': [3, 480, 640]}, 'observation.overhead': {'mean': [[[0.545862979096059]], [[0.5452054301949182]], [[0.5477746602359932]]], 'std': [[[0.14197105669614393]], [[0.1436179498089634]], [[0.14500463044962006]]], 'min': [[[0.0]], [[0.0]], [[0.0]]], 'max': [[[1.0]], [[1.0]], [[1.0]]], 'q01': [[[0.05494559589188223]], [[0.05129975818181184]], [[0.07399745843495421]]], 'q99': [[[0.7951740036832365]], [[0.8620514435976284]], [[0.9287900526200447]]], 'type': 'VISUAL', 'name': 'overhead', 'shape': [3, 480, 640]}, 'action': {'mean': [0.47757554785419337, -18.034652611064505, 15.59805635303217, 54.54463619743119, -49.21284246567293, 7.302610387584375], 'std': [29.091115680540447, 44.16383911983223, 44.81775979436524, 19.76430732442855, 8.799572987673075, 10.41276650889744], 'min': [-75.1649169921875, -100.0, -61.050724029541016, -6.849315166473389, -86.33736419677734, 0.0], 'max': [81.91695404052734, 72.79933166503906, 100.0, 93.32191467285156, -2.732527494430542, 42.88956069946289], 'q01': [-41.29065982641405, -100.0, -27.74406375139188, 24.550995379247542, -61.591176330648565, -1.000000013351432e-10], 'q99': [34.99315845909429, 38.49093164574106, 99.99756689137342, 82.60630455329574, -36.44491567201352, 26.6719993779817], 'type': 'ACTION', 'name': 'action', 'shape': [6]}}}
phase2_snapflow/wandb/run-20260807_093805-7i8cqirl/run-7i8cqirl.wandb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:efbbca0bfae236be9ed6fa201bfc416d6580af8f2908ee94eeaed7c673635e15
3
+ size 16777216