Add DECO model files
Browse files- .gitattributes +1 -0
- deco/deco_stage1_latest.pt +3 -0
- deco/deco_stage1_latest.ts +3 -0
- deco/deco_stage1_latest.ts.json +97 -0
.gitattributes
CHANGED
|
@@ -38,3 +38,4 @@ at_20260813_115524/wandb/offline-run-20260813_115535-v9gxoeo7/run-v9gxoeo7.wandb
|
|
| 38 |
pick_tube_01_06/at_rdp0106_full_20260816/logs.json.txt filter=lfs diff=lfs merge=lfs -text
|
| 39 |
pick_tube_01_06/at_rdp0106_full_20260816/wandb/offline-run-20260816_185254-oszq7iuv/run-oszq7iuv.wandb filter=lfs diff=lfs merge=lfs -text
|
| 40 |
pick_tube_01_06/ldp_rdp0106_ldp_b32w8_20260816/logs.json.txt filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 38 |
pick_tube_01_06/at_rdp0106_full_20260816/logs.json.txt filter=lfs diff=lfs merge=lfs -text
|
| 39 |
pick_tube_01_06/at_rdp0106_full_20260816/wandb/offline-run-20260816_185254-oszq7iuv/run-oszq7iuv.wandb filter=lfs diff=lfs merge=lfs -text
|
| 40 |
pick_tube_01_06/ldp_rdp0106_ldp_b32w8_20260816/logs.json.txt filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
deco/deco_stage1_latest.ts filter=lfs diff=lfs merge=lfs -text
|
deco/deco_stage1_latest.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c182b2e40b3808f45bfd8b7732b6c317a5e5b70eaa1ffa515c066f63e57ab5cf
|
| 3 |
+
size 977986431
|
deco/deco_stage1_latest.ts
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2123b42e2efd1d7fc634f45e46319bcbca2b25cedee4af8f95aa8abf1d32e550
|
| 3 |
+
size 322963489
|
deco/deco_stage1_latest.ts.json
ADDED
|
@@ -0,0 +1,97 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"camera_names": [
|
| 3 |
+
"observation.images.camera0",
|
| 4 |
+
"observation.images.camera1"
|
| 5 |
+
],
|
| 6 |
+
"dataset_id": "pick_tube_01_02_03_04_05_06@vision2-v1-val0.1-seed42",
|
| 7 |
+
"epoch": 60,
|
| 8 |
+
"expected_sample_hz": 30.0,
|
| 9 |
+
"format": "sudo-upstream-deco-stage1-torchscript-v1",
|
| 10 |
+
"inference_steps": 5,
|
| 11 |
+
"input": {
|
| 12 |
+
"images": [
|
| 13 |
+
1,
|
| 14 |
+
2,
|
| 15 |
+
3,
|
| 16 |
+
224,
|
| 17 |
+
224
|
| 18 |
+
],
|
| 19 |
+
"images_dtype": "float32",
|
| 20 |
+
"images_range": [
|
| 21 |
+
0.0,
|
| 22 |
+
1.0
|
| 23 |
+
],
|
| 24 |
+
"observation": [
|
| 25 |
+
1,
|
| 26 |
+
20
|
| 27 |
+
],
|
| 28 |
+
"observation_space": "raw source state; mapping and normalization embedded",
|
| 29 |
+
"state_columns": [
|
| 30 |
+
"robot0.relative_start.x",
|
| 31 |
+
"robot0.relative_start.y",
|
| 32 |
+
"robot0.relative_start.z",
|
| 33 |
+
"robot0.relative_start.rx",
|
| 34 |
+
"robot0.relative_start.ry",
|
| 35 |
+
"robot0.relative_start.rz",
|
| 36 |
+
"robot0.gripper_width",
|
| 37 |
+
"robot1.relative_start.x",
|
| 38 |
+
"robot1.relative_start.y",
|
| 39 |
+
"robot1.relative_start.z",
|
| 40 |
+
"robot1.relative_start.rx",
|
| 41 |
+
"robot1.relative_start.ry",
|
| 42 |
+
"robot1.relative_start.rz",
|
| 43 |
+
"robot1.gripper_width",
|
| 44 |
+
"left_relative_to_right.x",
|
| 45 |
+
"left_relative_to_right.y",
|
| 46 |
+
"left_relative_to_right.z",
|
| 47 |
+
"left_relative_to_right.rx",
|
| 48 |
+
"left_relative_to_right.ry",
|
| 49 |
+
"left_relative_to_right.rz"
|
| 50 |
+
],
|
| 51 |
+
"state_layout": "relative_start_pose6d_gripper_plus_left_relative_right"
|
| 52 |
+
},
|
| 53 |
+
"normalization": {
|
| 54 |
+
"embedded": true,
|
| 55 |
+
"statistics_source": "train_episodes_nonterminal_rows_once"
|
| 56 |
+
},
|
| 57 |
+
"output": {
|
| 58 |
+
"action": [
|
| 59 |
+
1,
|
| 60 |
+
32,
|
| 61 |
+
20
|
| 62 |
+
],
|
| 63 |
+
"action_columns": [
|
| 64 |
+
"robot0.delta.x",
|
| 65 |
+
"robot0.delta.y",
|
| 66 |
+
"robot0.delta.z",
|
| 67 |
+
"robot0.rotation_column_0.x",
|
| 68 |
+
"robot0.rotation_column_0.y",
|
| 69 |
+
"robot0.rotation_column_0.z",
|
| 70 |
+
"robot0.rotation_column_1.x",
|
| 71 |
+
"robot0.rotation_column_1.y",
|
| 72 |
+
"robot0.rotation_column_1.z",
|
| 73 |
+
"robot0.gripper_width",
|
| 74 |
+
"robot1.delta.x",
|
| 75 |
+
"robot1.delta.y",
|
| 76 |
+
"robot1.delta.z",
|
| 77 |
+
"robot1.rotation_column_0.x",
|
| 78 |
+
"robot1.rotation_column_0.y",
|
| 79 |
+
"robot1.rotation_column_0.z",
|
| 80 |
+
"robot1.rotation_column_1.x",
|
| 81 |
+
"robot1.rotation_column_1.y",
|
| 82 |
+
"robot1.rotation_column_1.z",
|
| 83 |
+
"robot1.gripper_width"
|
| 84 |
+
],
|
| 85 |
+
"action_mode": "tcp_delta_absolute_gripper",
|
| 86 |
+
"action_space": "denormalized per-arm TCP-frame delta xyz + Rotation-6D matrix columns + absolute gripper width",
|
| 87 |
+
"gripper_mode": "absolute",
|
| 88 |
+
"rotation_representation": "rotation_6d_matrix_columns",
|
| 89 |
+
"terminal_action_policy": "excluded"
|
| 90 |
+
},
|
| 91 |
+
"output_path": "/DATA/ljl/substage/deco_runs/pick_tube_vision_01_06_ddp2_20260825_002542/deco_stage1_latest.ts",
|
| 92 |
+
"source": "training_loop",
|
| 93 |
+
"stochastic": true,
|
| 94 |
+
"torchscript_sha256": "2123b42e2efd1d7fc634f45e46319bcbca2b25cedee4af8f95aa8abf1d32e550",
|
| 95 |
+
"upstream_model": "models.deco.deco.DECO",
|
| 96 |
+
"val_loss": 0.2389727109574071
|
| 97 |
+
}
|