Upload folder using huggingface_hub
Browse files- .gitattributes +3 -0
- phase1/phase1-epoch003/exports/openvino/manifest.json +218 -0
- phase1/phase1-epoch003/exports/openvino/pi05.bin +3 -0
- phase1/phase1-epoch003/exports/openvino/pi05.xml +3 -0
- phase1/phase1-epoch003/exports/openvino/tokenizer.bin +3 -0
- phase1/phase1-epoch003/exports/openvino/tokenizer.xml +790 -0
- phase1/phase1-epoch005/exports/openvino/manifest.json +218 -0
- phase1/phase1-epoch005/exports/openvino/pi05.bin +3 -0
- phase1/phase1-epoch005/exports/openvino/pi05.xml +3 -0
- phase1/phase1-epoch005/exports/openvino/tokenizer.bin +3 -0
- phase1/phase1-epoch005/exports/openvino/tokenizer.xml +790 -0
- phase1/wandb/debug-internal.log +0 -0
- phase1/wandb/debug.log +26 -0
- phase1/wandb/run-20260806_134958-uwqkxwz8/files/config.yaml +562 -0
- phase1/wandb/run-20260806_134958-uwqkxwz8/files/output.log +0 -0
- phase1/wandb/run-20260806_134958-uwqkxwz8/files/requirements.txt +197 -0
- phase1/wandb/run-20260806_134958-uwqkxwz8/files/wandb-metadata.json +71 -0
- phase1/wandb/run-20260806_134958-uwqkxwz8/files/wandb-summary.json +1 -0
- phase1/wandb/run-20260806_134958-uwqkxwz8/logs/debug-core.log +27 -0
- phase1/wandb/run-20260806_134958-uwqkxwz8/logs/debug-internal.log +0 -0
- phase1/wandb/run-20260806_134958-uwqkxwz8/logs/debug.log +26 -0
- phase1/wandb/run-20260806_134958-uwqkxwz8/run-uwqkxwz8.wandb +3 -0
.gitattributes
CHANGED
|
@@ -35,3 +35,6 @@ saved_model/**/* 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
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|
| 38 |
+
phase1/phase1-epoch003/exports/openvino/pi05.xml filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
phase1/phase1-epoch005/exports/openvino/pi05.xml filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
phase1/wandb/run-20260806_134958-uwqkxwz8/run-uwqkxwz8.wandb filter=lfs diff=lfs merge=lfs -text
|
phase1/phase1-epoch003/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 |
+
}
|
phase1/phase1-epoch003/exports/openvino/pi05.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:065f9c9314559d154d8369a4fc0d9a481034c0812410c339cb3ade2493fc4938
|
| 3 |
+
size 6254309792
|
phase1/phase1-epoch003/exports/openvino/pi05.xml
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f9c1c600f766311b5bc30a9328e3f78c6296f4924cefbf68d61986fda7c0825c
|
| 3 |
+
size 11813377
|
phase1/phase1-epoch003/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
|
phase1/phase1-epoch003/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_164947" type="Parameter" version="opset1">
|
| 5 |
+
<data shape="?" element_type="string" />
|
| 6 |
+
<output>
|
| 7 |
+
<port id="0" precision="STRING" names="Parameter_164947">
|
| 8 |
+
<dim>-1</dim>
|
| 9 |
+
</port>
|
| 10 |
+
</output>
|
| 11 |
+
</layer>
|
| 12 |
+
<layer id="1" name="Constant_165066" 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_165067" 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_165068" 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_164953" 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_164948" 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_164949" 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_164950" 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_164951" 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_164952" 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_164954" 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_164955" 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_164956" 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_164957" 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_164958" 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_164959" 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_164960" 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_165024" 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_165025" 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_165027" 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_165029" 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_165030" 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_165032" 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_165033" 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_165035" 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_165037" 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_165039" 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_165047" 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_165049" 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_165051" 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_165053" 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_165055" 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_165057" 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_165041" 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_165043" 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_165045" 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_165058" 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_165059" type="BPETokenizer" version="extension">
|
| 416 |
+
<data unk_token="<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_165060" 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_165062" 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_165064" 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_165065" 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_165069" 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_165070" 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_165071" 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_165072" 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_165073" 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_165074" 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_165074.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_165073.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_165077" 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_165079" 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="{"type": "TemplateProcessing", "single": [{"SpecialToken": {"id": "<bos>", "type_id": 0}}, {"Sequence": {"id": "A", "type_id": 0}}], "pair": [{"SpecialToken": {"id": "<bos>", "type_id": 0}}, {"Sequence": {"id": "A", "type_id": 0}}, {"SpecialToken": {"id": "<bos>", "type_id": 1}}, {"Sequence": {"id": "B", "type_id": 1}}], "special_tokens": {"<bos>": {"id": "<bos>", "ids": [2], "tokens": ["<bos>"]}}}" />
|
| 777 |
+
<original_tokenizer_class value="<class 'transformers.models.gemma.tokenization_gemma.GemmaTokenizer'>" />
|
| 778 |
+
<pad_token_id value="0" />
|
| 779 |
+
<processed_post_processor_template value="{"single": {"ids": [2, -1], "type_ids": [0, 0]}, "pair": {"ids": [2, -1, 2, -2], "type_ids": [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>
|
phase1/phase1-epoch005/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 |
+
}
|
phase1/phase1-epoch005/exports/openvino/pi05.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:61b407fb9bd3fb477eed95bb630f14900fbfc3df03a653eb91bfae8a90b9693d
|
| 3 |
+
size 6254309792
|
phase1/phase1-epoch005/exports/openvino/pi05.xml
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3c1cb3ffd2931d60a0f3f978e0883cb11ac091ef20546987fe63013a7cf90467
|
| 3 |
+
size 11814801
|
phase1/phase1-epoch005/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
|
phase1/phase1-epoch005/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_330058" type="Parameter" version="opset1">
|
| 5 |
+
<data shape="?" element_type="string" />
|
| 6 |
+
<output>
|
| 7 |
+
<port id="0" precision="STRING" names="Parameter_330058">
|
| 8 |
+
<dim>-1</dim>
|
| 9 |
+
</port>
|
| 10 |
+
</output>
|
| 11 |
+
</layer>
|
| 12 |
+
<layer id="1" name="Constant_330177" 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_330178" 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_330179" 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_330064" 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_330059" 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_330060" 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_330061" 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_330062" 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_330063" 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_330065" 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_330066" 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_330067" 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_330068" 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_330069" 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_330070" 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_330071" 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_330135" 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_330136" 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_330138" 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_330140" 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_330141" 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_330143" 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_330144" 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_330146" 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_330148" 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_330150" 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_330158" 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_330160" 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_330162" 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_330164" 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_330166" 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_330168" 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_330152" 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_330154" 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_330156" 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_330169" 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_330170" type="BPETokenizer" version="extension">
|
| 416 |
+
<data unk_token="<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_330171" 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_330173" 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_330175" 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_330176" 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_330180" 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_330181" 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_330182" 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_330183" 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_330184" 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_330185" 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_330185.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_330184.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_330188" 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_330190" 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="{"type": "TemplateProcessing", "single": [{"SpecialToken": {"id": "<bos>", "type_id": 0}}, {"Sequence": {"id": "A", "type_id": 0}}], "pair": [{"SpecialToken": {"id": "<bos>", "type_id": 0}}, {"Sequence": {"id": "A", "type_id": 0}}, {"SpecialToken": {"id": "<bos>", "type_id": 1}}, {"Sequence": {"id": "B", "type_id": 1}}], "special_tokens": {"<bos>": {"id": "<bos>", "ids": [2], "tokens": ["<bos>"]}}}" />
|
| 777 |
+
<original_tokenizer_class value="<class 'transformers.models.gemma.tokenization_gemma.GemmaTokenizer'>" />
|
| 778 |
+
<pad_token_id value="0" />
|
| 779 |
+
<processed_post_processor_template value="{"single": {"ids": [2, -1], "type_ids": [0, 0]}, "pair": {"ids": [2, -1, 2, -2], "type_ids": [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>
|
phase1/wandb/debug-internal.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
phase1/wandb/debug.log
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_setup.py:_flush():81] Current SDK version is 0.28.1
|
| 2 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_setup.py:_flush():81] Configure stats pid to 4075244
|
| 3 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_setup.py:_flush():81] Loading settings from environment variables
|
| 4 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:setup_run_log_directory():729] Logging user logs to /mnt/data/experiments/snapflow-multi-obj/phase1/wandb/run-20260806_134958-uwqkxwz8/logs/debug.log
|
| 5 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:setup_run_log_directory():730] Logging internal logs to /mnt/data/experiments/snapflow-multi-obj/phase1/wandb/run-20260806_134958-uwqkxwz8/logs/debug-internal.log
|
| 6 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:init():772] calling init triggers
|
| 7 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:init():777] wandb.init called with sweep_config: {}
|
| 8 |
+
config: {'_wandb': {}}
|
| 9 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:init():820] starting backend
|
| 10 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:init():835] sending inform_init request
|
| 11 |
+
2026-08-06 13:49:58,796 INFO MainThread:4075244 [wandb_init.py:init():840] backend started and connected
|
| 12 |
+
2026-08-06 13:49:58,799 INFO MainThread:4075244 [wandb_init.py:init():910] updated telemetry
|
| 13 |
+
2026-08-06 13:49:58,809 INFO MainThread:4075244 [wandb_init.py:init():933] communicating run to backend with 90.0 second timeout
|
| 14 |
+
2026-08-06 13:49:59,540 INFO MainThread:4075244 [wandb_init.py:init():978] starting run threads in backend
|
| 15 |
+
2026-08-06 13:49:59,644 INFO MainThread:4075244 [wandb_run.py:_console_start():2621] atexit reg
|
| 16 |
+
2026-08-06 13:49:59,644 INFO MainThread:4075244 [wandb_run.py:_redirect():2471] redirect: wrap_raw
|
| 17 |
+
2026-08-06 13:49:59,644 INFO MainThread:4075244 [wandb_run.py:_redirect():2540] Wrapping output streams.
|
| 18 |
+
2026-08-06 13:49:59,644 INFO MainThread:4075244 [wandb_run.py:_redirect():2563] Redirects installed.
|
| 19 |
+
2026-08-06 13:49:59,647 INFO MainThread:4075244 [wandb_init.py:init():1016] run started, returning control to user process
|
| 20 |
+
2026-08-06 13:49:59,648 INFO MainThread:4075244 [wandb_run.py:_config_callback():1346] config_cb None None {'phase': 'phase1', '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': 7, 'train_frames': 86093, 'global_batch': 16, 'steps_per_epoch': 5380, 'total_steps': 37660, 'warmup_steps': 1883, 'warmup_frac': 0.05, 'snapflow_alpha': 0.5, 'snapflow_lambda': 0.1, 'snapflow_steps': 1}
|
| 21 |
+
2026-08-06 13:50:02,686 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': False, 'snapflow_alpha': 0.5, 'snapflow_lambda': 1.0, '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': False, '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': 1883, 'scheduler_decay_steps': None, 'scheduler_decay_lr': 2.5e-06, 'dataset_stats': {'observation.images.base_0_rgb': {'name': 'observation.images.base_0_rgb', 'shape': [3, 224, 224], 'type': 'VISUAL', 'mean': [0.0, 0.0, 0.0], 'std': [1.0, 1.0, 1.0], 'q01': [-1.0, -1.0, -1.0], 'q99': [1.0, 1.0, 1.0]}, 'observation.images.left_wrist_0_rgb': {'name': 'observation.images.left_wrist_0_rgb', 'shape': [3, 224, 224], 'type': 'VISUAL', 'mean': [0.0, 0.0, 0.0], 'std': [1.0, 1.0, 1.0], 'q01': [-1.0, -1.0, -1.0], 'q99': [1.0, 1.0, 1.0]}, 'observation.images.right_wrist_0_rgb': {'name': 'observation.images.right_wrist_0_rgb', 'shape': [3, 224, 224], 'type': 'VISUAL', 'mean': [0.0, 0.0, 0.0], 'std': [1.0, 1.0, 1.0], 'q01': [-1.0, -1.0, -1.0], 'q99': [1.0, 1.0, 1.0]}, 'observation.state': {'name': 'observation.state', 'shape': [32], 'type': 'STATE', 'mean': [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], 'std': [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], 'q01': [-1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0], 'q99': [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]}, 'action': {'name': 'action', 'shape': [32], 'type': 'ACTION', 'mean': [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], 'std': [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], 'q01': [-1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0], 'q99': [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]}}}
|
| 22 |
+
2026-08-06 23:54:26,913 INFO MainThread:4075244 [wandb_run.py:_finish():2383] finishing run daankrol/physicalai_snapflow/uwqkxwz8
|
| 23 |
+
2026-08-06 23:54:26,914 INFO MainThread:4075244 [wandb_run.py:_atexit_cleanup():2588] got exitcode: 0
|
| 24 |
+
2026-08-06 23:54:26,914 INFO MainThread:4075244 [wandb_run.py:_restore():2570] restore
|
| 25 |
+
2026-08-06 23:54:26,914 INFO MainThread:4075244 [wandb_run.py:_restore():2576] restore done
|
| 26 |
+
2026-08-06 23:54:28,757 INFO MainThread:4075244 [wandb_run.py:_footer_sync_info():3993] logging synced files
|
phase1/wandb/run-20260806_134958-uwqkxwz8/files/config.yaml
ADDED
|
@@ -0,0 +1,562 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
_wandb:
|
| 2 |
+
value:
|
| 3 |
+
cli_version: 0.28.1
|
| 4 |
+
e:
|
| 5 |
+
sfvgepz6a7tk4syemnvwssotyebxp7ch:
|
| 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: "1709108813824"
|
| 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/phase1
|
| 59 |
+
startedAt: "2026-08-06T13:49:58.671065Z"
|
| 60 |
+
writerId: sfvgepz6a7tk4syemnvwssotyebxp7ch
|
| 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 |
+
mean:
|
| 124 |
+
- 0
|
| 125 |
+
- 0
|
| 126 |
+
- 0
|
| 127 |
+
- 0
|
| 128 |
+
- 0
|
| 129 |
+
- 0
|
| 130 |
+
- 0
|
| 131 |
+
- 0
|
| 132 |
+
- 0
|
| 133 |
+
- 0
|
| 134 |
+
- 0
|
| 135 |
+
- 0
|
| 136 |
+
- 0
|
| 137 |
+
- 0
|
| 138 |
+
- 0
|
| 139 |
+
- 0
|
| 140 |
+
- 0
|
| 141 |
+
- 0
|
| 142 |
+
- 0
|
| 143 |
+
- 0
|
| 144 |
+
- 0
|
| 145 |
+
- 0
|
| 146 |
+
- 0
|
| 147 |
+
- 0
|
| 148 |
+
- 0
|
| 149 |
+
- 0
|
| 150 |
+
- 0
|
| 151 |
+
- 0
|
| 152 |
+
- 0
|
| 153 |
+
- 0
|
| 154 |
+
- 0
|
| 155 |
+
- 0
|
| 156 |
+
name: action
|
| 157 |
+
q01:
|
| 158 |
+
- -1
|
| 159 |
+
- -1
|
| 160 |
+
- -1
|
| 161 |
+
- -1
|
| 162 |
+
- -1
|
| 163 |
+
- -1
|
| 164 |
+
- -1
|
| 165 |
+
- -1
|
| 166 |
+
- -1
|
| 167 |
+
- -1
|
| 168 |
+
- -1
|
| 169 |
+
- -1
|
| 170 |
+
- -1
|
| 171 |
+
- -1
|
| 172 |
+
- -1
|
| 173 |
+
- -1
|
| 174 |
+
- -1
|
| 175 |
+
- -1
|
| 176 |
+
- -1
|
| 177 |
+
- -1
|
| 178 |
+
- -1
|
| 179 |
+
- -1
|
| 180 |
+
- -1
|
| 181 |
+
- -1
|
| 182 |
+
- -1
|
| 183 |
+
- -1
|
| 184 |
+
- -1
|
| 185 |
+
- -1
|
| 186 |
+
- -1
|
| 187 |
+
- -1
|
| 188 |
+
- -1
|
| 189 |
+
- -1
|
| 190 |
+
q99:
|
| 191 |
+
- 1
|
| 192 |
+
- 1
|
| 193 |
+
- 1
|
| 194 |
+
- 1
|
| 195 |
+
- 1
|
| 196 |
+
- 1
|
| 197 |
+
- 1
|
| 198 |
+
- 1
|
| 199 |
+
- 1
|
| 200 |
+
- 1
|
| 201 |
+
- 1
|
| 202 |
+
- 1
|
| 203 |
+
- 1
|
| 204 |
+
- 1
|
| 205 |
+
- 1
|
| 206 |
+
- 1
|
| 207 |
+
- 1
|
| 208 |
+
- 1
|
| 209 |
+
- 1
|
| 210 |
+
- 1
|
| 211 |
+
- 1
|
| 212 |
+
- 1
|
| 213 |
+
- 1
|
| 214 |
+
- 1
|
| 215 |
+
- 1
|
| 216 |
+
- 1
|
| 217 |
+
- 1
|
| 218 |
+
- 1
|
| 219 |
+
- 1
|
| 220 |
+
- 1
|
| 221 |
+
- 1
|
| 222 |
+
- 1
|
| 223 |
+
shape:
|
| 224 |
+
- 32
|
| 225 |
+
std:
|
| 226 |
+
- 1
|
| 227 |
+
- 1
|
| 228 |
+
- 1
|
| 229 |
+
- 1
|
| 230 |
+
- 1
|
| 231 |
+
- 1
|
| 232 |
+
- 1
|
| 233 |
+
- 1
|
| 234 |
+
- 1
|
| 235 |
+
- 1
|
| 236 |
+
- 1
|
| 237 |
+
- 1
|
| 238 |
+
- 1
|
| 239 |
+
- 1
|
| 240 |
+
- 1
|
| 241 |
+
- 1
|
| 242 |
+
- 1
|
| 243 |
+
- 1
|
| 244 |
+
- 1
|
| 245 |
+
- 1
|
| 246 |
+
- 1
|
| 247 |
+
- 1
|
| 248 |
+
- 1
|
| 249 |
+
- 1
|
| 250 |
+
- 1
|
| 251 |
+
- 1
|
| 252 |
+
- 1
|
| 253 |
+
- 1
|
| 254 |
+
- 1
|
| 255 |
+
- 1
|
| 256 |
+
- 1
|
| 257 |
+
- 1
|
| 258 |
+
type: ACTION
|
| 259 |
+
observation.images.base_0_rgb:
|
| 260 |
+
mean:
|
| 261 |
+
- 0
|
| 262 |
+
- 0
|
| 263 |
+
- 0
|
| 264 |
+
name: observation.images.base_0_rgb
|
| 265 |
+
q01:
|
| 266 |
+
- -1
|
| 267 |
+
- -1
|
| 268 |
+
- -1
|
| 269 |
+
q99:
|
| 270 |
+
- 1
|
| 271 |
+
- 1
|
| 272 |
+
- 1
|
| 273 |
+
shape:
|
| 274 |
+
- 3
|
| 275 |
+
- 224
|
| 276 |
+
- 224
|
| 277 |
+
std:
|
| 278 |
+
- 1
|
| 279 |
+
- 1
|
| 280 |
+
- 1
|
| 281 |
+
type: VISUAL
|
| 282 |
+
observation.images.left_wrist_0_rgb:
|
| 283 |
+
mean:
|
| 284 |
+
- 0
|
| 285 |
+
- 0
|
| 286 |
+
- 0
|
| 287 |
+
name: observation.images.left_wrist_0_rgb
|
| 288 |
+
q01:
|
| 289 |
+
- -1
|
| 290 |
+
- -1
|
| 291 |
+
- -1
|
| 292 |
+
q99:
|
| 293 |
+
- 1
|
| 294 |
+
- 1
|
| 295 |
+
- 1
|
| 296 |
+
shape:
|
| 297 |
+
- 3
|
| 298 |
+
- 224
|
| 299 |
+
- 224
|
| 300 |
+
std:
|
| 301 |
+
- 1
|
| 302 |
+
- 1
|
| 303 |
+
- 1
|
| 304 |
+
type: VISUAL
|
| 305 |
+
observation.images.right_wrist_0_rgb:
|
| 306 |
+
mean:
|
| 307 |
+
- 0
|
| 308 |
+
- 0
|
| 309 |
+
- 0
|
| 310 |
+
name: observation.images.right_wrist_0_rgb
|
| 311 |
+
q01:
|
| 312 |
+
- -1
|
| 313 |
+
- -1
|
| 314 |
+
- -1
|
| 315 |
+
q99:
|
| 316 |
+
- 1
|
| 317 |
+
- 1
|
| 318 |
+
- 1
|
| 319 |
+
shape:
|
| 320 |
+
- 3
|
| 321 |
+
- 224
|
| 322 |
+
- 224
|
| 323 |
+
std:
|
| 324 |
+
- 1
|
| 325 |
+
- 1
|
| 326 |
+
- 1
|
| 327 |
+
type: VISUAL
|
| 328 |
+
observation.state:
|
| 329 |
+
mean:
|
| 330 |
+
- 0
|
| 331 |
+
- 0
|
| 332 |
+
- 0
|
| 333 |
+
- 0
|
| 334 |
+
- 0
|
| 335 |
+
- 0
|
| 336 |
+
- 0
|
| 337 |
+
- 0
|
| 338 |
+
- 0
|
| 339 |
+
- 0
|
| 340 |
+
- 0
|
| 341 |
+
- 0
|
| 342 |
+
- 0
|
| 343 |
+
- 0
|
| 344 |
+
- 0
|
| 345 |
+
- 0
|
| 346 |
+
- 0
|
| 347 |
+
- 0
|
| 348 |
+
- 0
|
| 349 |
+
- 0
|
| 350 |
+
- 0
|
| 351 |
+
- 0
|
| 352 |
+
- 0
|
| 353 |
+
- 0
|
| 354 |
+
- 0
|
| 355 |
+
- 0
|
| 356 |
+
- 0
|
| 357 |
+
- 0
|
| 358 |
+
- 0
|
| 359 |
+
- 0
|
| 360 |
+
- 0
|
| 361 |
+
- 0
|
| 362 |
+
name: observation.state
|
| 363 |
+
q01:
|
| 364 |
+
- -1
|
| 365 |
+
- -1
|
| 366 |
+
- -1
|
| 367 |
+
- -1
|
| 368 |
+
- -1
|
| 369 |
+
- -1
|
| 370 |
+
- -1
|
| 371 |
+
- -1
|
| 372 |
+
- -1
|
| 373 |
+
- -1
|
| 374 |
+
- -1
|
| 375 |
+
- -1
|
| 376 |
+
- -1
|
| 377 |
+
- -1
|
| 378 |
+
- -1
|
| 379 |
+
- -1
|
| 380 |
+
- -1
|
| 381 |
+
- -1
|
| 382 |
+
- -1
|
| 383 |
+
- -1
|
| 384 |
+
- -1
|
| 385 |
+
- -1
|
| 386 |
+
- -1
|
| 387 |
+
- -1
|
| 388 |
+
- -1
|
| 389 |
+
- -1
|
| 390 |
+
- -1
|
| 391 |
+
- -1
|
| 392 |
+
- -1
|
| 393 |
+
- -1
|
| 394 |
+
- -1
|
| 395 |
+
- -1
|
| 396 |
+
q99:
|
| 397 |
+
- 1
|
| 398 |
+
- 1
|
| 399 |
+
- 1
|
| 400 |
+
- 1
|
| 401 |
+
- 1
|
| 402 |
+
- 1
|
| 403 |
+
- 1
|
| 404 |
+
- 1
|
| 405 |
+
- 1
|
| 406 |
+
- 1
|
| 407 |
+
- 1
|
| 408 |
+
- 1
|
| 409 |
+
- 1
|
| 410 |
+
- 1
|
| 411 |
+
- 1
|
| 412 |
+
- 1
|
| 413 |
+
- 1
|
| 414 |
+
- 1
|
| 415 |
+
- 1
|
| 416 |
+
- 1
|
| 417 |
+
- 1
|
| 418 |
+
- 1
|
| 419 |
+
- 1
|
| 420 |
+
- 1
|
| 421 |
+
- 1
|
| 422 |
+
- 1
|
| 423 |
+
- 1
|
| 424 |
+
- 1
|
| 425 |
+
- 1
|
| 426 |
+
- 1
|
| 427 |
+
- 1
|
| 428 |
+
- 1
|
| 429 |
+
shape:
|
| 430 |
+
- 32
|
| 431 |
+
std:
|
| 432 |
+
- 1
|
| 433 |
+
- 1
|
| 434 |
+
- 1
|
| 435 |
+
- 1
|
| 436 |
+
- 1
|
| 437 |
+
- 1
|
| 438 |
+
- 1
|
| 439 |
+
- 1
|
| 440 |
+
- 1
|
| 441 |
+
- 1
|
| 442 |
+
- 1
|
| 443 |
+
- 1
|
| 444 |
+
- 1
|
| 445 |
+
- 1
|
| 446 |
+
- 1
|
| 447 |
+
- 1
|
| 448 |
+
- 1
|
| 449 |
+
- 1
|
| 450 |
+
- 1
|
| 451 |
+
- 1
|
| 452 |
+
- 1
|
| 453 |
+
- 1
|
| 454 |
+
- 1
|
| 455 |
+
- 1
|
| 456 |
+
- 1
|
| 457 |
+
- 1
|
| 458 |
+
- 1
|
| 459 |
+
- 1
|
| 460 |
+
- 1
|
| 461 |
+
- 1
|
| 462 |
+
- 1
|
| 463 |
+
- 1
|
| 464 |
+
type: STATE
|
| 465 |
+
devices:
|
| 466 |
+
value: 1
|
| 467 |
+
dtype:
|
| 468 |
+
value: bfloat16
|
| 469 |
+
empty_cameras:
|
| 470 |
+
value: 0
|
| 471 |
+
epochs:
|
| 472 |
+
value: 7
|
| 473 |
+
freeze_vision_encoder:
|
| 474 |
+
value: false
|
| 475 |
+
global_batch:
|
| 476 |
+
value: 16
|
| 477 |
+
gradient_checkpointing:
|
| 478 |
+
value: true
|
| 479 |
+
image_resolution:
|
| 480 |
+
value:
|
| 481 |
+
- 224
|
| 482 |
+
- 224
|
| 483 |
+
max_action_dim:
|
| 484 |
+
value: 32
|
| 485 |
+
max_period:
|
| 486 |
+
value: 4
|
| 487 |
+
max_state_dim:
|
| 488 |
+
value: 32
|
| 489 |
+
min_period:
|
| 490 |
+
value: 0.004
|
| 491 |
+
n_action_steps:
|
| 492 |
+
value: 50
|
| 493 |
+
n_obs_steps:
|
| 494 |
+
value: 1
|
| 495 |
+
normalization_mode:
|
| 496 |
+
value: QUANTILES
|
| 497 |
+
num_inference_steps:
|
| 498 |
+
value: 10
|
| 499 |
+
optimizer_betas:
|
| 500 |
+
value:
|
| 501 |
+
- 0.9
|
| 502 |
+
- 0.95
|
| 503 |
+
optimizer_eps:
|
| 504 |
+
value: 1e-08
|
| 505 |
+
optimizer_grad_clip_norm:
|
| 506 |
+
value: 1
|
| 507 |
+
optimizer_lr:
|
| 508 |
+
value: 2.5e-05
|
| 509 |
+
optimizer_weight_decay:
|
| 510 |
+
value: 0.01
|
| 511 |
+
paligemma_variant:
|
| 512 |
+
value: gemma_2b
|
| 513 |
+
phase:
|
| 514 |
+
value: phase1
|
| 515 |
+
pretrained:
|
| 516 |
+
value: lerobot/pi05_base
|
| 517 |
+
scheduler_decay_lr:
|
| 518 |
+
value: 2.5e-06
|
| 519 |
+
scheduler_decay_steps:
|
| 520 |
+
value: null
|
| 521 |
+
scheduler_warmup_steps:
|
| 522 |
+
value: 1883
|
| 523 |
+
seed:
|
| 524 |
+
value: 42
|
| 525 |
+
snapflow_alpha:
|
| 526 |
+
value: 0.5
|
| 527 |
+
snapflow_enabled:
|
| 528 |
+
value: false
|
| 529 |
+
snapflow_lambda:
|
| 530 |
+
value: 1
|
| 531 |
+
snapflow_num_inference_steps:
|
| 532 |
+
value: 1
|
| 533 |
+
snapflow_steps:
|
| 534 |
+
value: 1
|
| 535 |
+
steps_per_epoch:
|
| 536 |
+
value: 5380
|
| 537 |
+
time_sampling_beta_alpha:
|
| 538 |
+
value: 1.5
|
| 539 |
+
time_sampling_beta_beta:
|
| 540 |
+
value: 1
|
| 541 |
+
time_sampling_offset:
|
| 542 |
+
value: 0.001
|
| 543 |
+
time_sampling_scale:
|
| 544 |
+
value: 0.999
|
| 545 |
+
tokenizer_max_length:
|
| 546 |
+
value: 200
|
| 547 |
+
total_steps:
|
| 548 |
+
value: 37660
|
| 549 |
+
train_expert_only:
|
| 550 |
+
value: false
|
| 551 |
+
train_frames:
|
| 552 |
+
value: 86093
|
| 553 |
+
use_random_input_noise:
|
| 554 |
+
value: true
|
| 555 |
+
val_split:
|
| 556 |
+
value: 0.1
|
| 557 |
+
val_split_seed:
|
| 558 |
+
value: 42
|
| 559 |
+
warmup_frac:
|
| 560 |
+
value: 0.05
|
| 561 |
+
warmup_steps:
|
| 562 |
+
value: 1883
|
phase1/wandb/run-20260806_134958-uwqkxwz8/files/output.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
phase1/wandb/run-20260806_134958-uwqkxwz8/files/requirements.txt
ADDED
|
@@ -0,0 +1,197 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|
phase1/wandb/run-20260806_134958-uwqkxwz8/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-06T13:49:58.671065Z",
|
| 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/phase1",
|
| 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": "1709108813824"
|
| 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": "sfvgepz6a7tk4syemnvwssotyebxp7ch"
|
| 71 |
+
}
|
phase1/wandb/run-20260806_134958-uwqkxwz8/files/wandb-summary.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"_timestamp":1.786060453114514e+09,"_runtime":36267,"_step":1512,"_wandb":{"runtime":36267},"trainer/global_step":37659,"lr-AdamW":2.5000047363848206e-06,"epoch":6,"val/loss":0.09501674771308899,"train/loss":0.017482014372944832}
|
phase1/wandb/run-20260806_134958-uwqkxwz8/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"}
|
phase1/wandb/run-20260806_134958-uwqkxwz8/logs/debug-internal.log
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
phase1/wandb/run-20260806_134958-uwqkxwz8/logs/debug.log
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_setup.py:_flush():81] Current SDK version is 0.28.1
|
| 2 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_setup.py:_flush():81] Configure stats pid to 4075244
|
| 3 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_setup.py:_flush():81] Loading settings from environment variables
|
| 4 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:setup_run_log_directory():729] Logging user logs to /mnt/data/experiments/snapflow-multi-obj/phase1/wandb/run-20260806_134958-uwqkxwz8/logs/debug.log
|
| 5 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:setup_run_log_directory():730] Logging internal logs to /mnt/data/experiments/snapflow-multi-obj/phase1/wandb/run-20260806_134958-uwqkxwz8/logs/debug-internal.log
|
| 6 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:init():772] calling init triggers
|
| 7 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:init():777] wandb.init called with sweep_config: {}
|
| 8 |
+
config: {'_wandb': {}}
|
| 9 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:init():820] starting backend
|
| 10 |
+
2026-08-06 13:49:58,672 INFO MainThread:4075244 [wandb_init.py:init():835] sending inform_init request
|
| 11 |
+
2026-08-06 13:49:58,796 INFO MainThread:4075244 [wandb_init.py:init():840] backend started and connected
|
| 12 |
+
2026-08-06 13:49:58,799 INFO MainThread:4075244 [wandb_init.py:init():910] updated telemetry
|
| 13 |
+
2026-08-06 13:49:58,809 INFO MainThread:4075244 [wandb_init.py:init():933] communicating run to backend with 90.0 second timeout
|
| 14 |
+
2026-08-06 13:49:59,540 INFO MainThread:4075244 [wandb_init.py:init():978] starting run threads in backend
|
| 15 |
+
2026-08-06 13:49:59,644 INFO MainThread:4075244 [wandb_run.py:_console_start():2621] atexit reg
|
| 16 |
+
2026-08-06 13:49:59,644 INFO MainThread:4075244 [wandb_run.py:_redirect():2471] redirect: wrap_raw
|
| 17 |
+
2026-08-06 13:49:59,644 INFO MainThread:4075244 [wandb_run.py:_redirect():2540] Wrapping output streams.
|
| 18 |
+
2026-08-06 13:49:59,644 INFO MainThread:4075244 [wandb_run.py:_redirect():2563] Redirects installed.
|
| 19 |
+
2026-08-06 13:49:59,647 INFO MainThread:4075244 [wandb_init.py:init():1016] run started, returning control to user process
|
| 20 |
+
2026-08-06 13:49:59,648 INFO MainThread:4075244 [wandb_run.py:_config_callback():1346] config_cb None None {'phase': 'phase1', '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': 7, 'train_frames': 86093, 'global_batch': 16, 'steps_per_epoch': 5380, 'total_steps': 37660, 'warmup_steps': 1883, 'warmup_frac': 0.05, 'snapflow_alpha': 0.5, 'snapflow_lambda': 0.1, 'snapflow_steps': 1}
|
| 21 |
+
2026-08-06 13:50:02,686 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': False, 'snapflow_alpha': 0.5, 'snapflow_lambda': 1.0, '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': False, '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': 1883, 'scheduler_decay_steps': None, 'scheduler_decay_lr': 2.5e-06, 'dataset_stats': {'observation.images.base_0_rgb': {'name': 'observation.images.base_0_rgb', 'shape': [3, 224, 224], 'type': 'VISUAL', 'mean': [0.0, 0.0, 0.0], 'std': [1.0, 1.0, 1.0], 'q01': [-1.0, -1.0, -1.0], 'q99': [1.0, 1.0, 1.0]}, 'observation.images.left_wrist_0_rgb': {'name': 'observation.images.left_wrist_0_rgb', 'shape': [3, 224, 224], 'type': 'VISUAL', 'mean': [0.0, 0.0, 0.0], 'std': [1.0, 1.0, 1.0], 'q01': [-1.0, -1.0, -1.0], 'q99': [1.0, 1.0, 1.0]}, 'observation.images.right_wrist_0_rgb': {'name': 'observation.images.right_wrist_0_rgb', 'shape': [3, 224, 224], 'type': 'VISUAL', 'mean': [0.0, 0.0, 0.0], 'std': [1.0, 1.0, 1.0], 'q01': [-1.0, -1.0, -1.0], 'q99': [1.0, 1.0, 1.0]}, 'observation.state': {'name': 'observation.state', 'shape': [32], 'type': 'STATE', 'mean': [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], 'std': [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], 'q01': [-1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0], 'q99': [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]}, 'action': {'name': 'action', 'shape': [32], 'type': 'ACTION', 'mean': [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], 'std': [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], 'q01': [-1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0], 'q99': [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]}}}
|
| 22 |
+
2026-08-06 23:54:26,913 INFO MainThread:4075244 [wandb_run.py:_finish():2383] finishing run daankrol/physicalai_snapflow/uwqkxwz8
|
| 23 |
+
2026-08-06 23:54:26,914 INFO MainThread:4075244 [wandb_run.py:_atexit_cleanup():2588] got exitcode: 0
|
| 24 |
+
2026-08-06 23:54:26,914 INFO MainThread:4075244 [wandb_run.py:_restore():2570] restore
|
| 25 |
+
2026-08-06 23:54:26,914 INFO MainThread:4075244 [wandb_run.py:_restore():2576] restore done
|
| 26 |
+
2026-08-06 23:54:28,757 INFO MainThread:4075244 [wandb_run.py:_footer_sync_info():3993] logging synced files
|
phase1/wandb/run-20260806_134958-uwqkxwz8/run-uwqkxwz8.wandb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4111375b62b6698a65ca5132eb8161721300c85e5c214a45cdfeef034fe02b78
|
| 3 |
+
size 37952863
|