Add delta_pose checkpoint (cubes T1/clean)
Browse files- delta_pose/config.json +123 -0
- delta_pose/dataset_stats.pkl +3 -0
- delta_pose/policy_best.ckpt +3 -0
delta_pose/config.json
ADDED
|
@@ -0,0 +1,123 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dataset": "dataset/quest_l2_cubes_pilot",
|
| 3 |
+
"instruction_id": "T1",
|
| 4 |
+
"complexity": "clean",
|
| 5 |
+
"chunk_size": 50,
|
| 6 |
+
"state_dim": 7,
|
| 7 |
+
"camera_names": [
|
| 8 |
+
"third_camera",
|
| 9 |
+
"wrist_camera"
|
| 10 |
+
],
|
| 11 |
+
"image_size": 240,
|
| 12 |
+
"kl_weight": 10.0,
|
| 13 |
+
"action_space": "tcp",
|
| 14 |
+
"instruction": "Pick the left red block and place it in the bin",
|
| 15 |
+
"train_episodes": [
|
| 16 |
+
3,
|
| 17 |
+
200,
|
| 18 |
+
190,
|
| 19 |
+
186,
|
| 20 |
+
192,
|
| 21 |
+
1,
|
| 22 |
+
177,
|
| 23 |
+
213,
|
| 24 |
+
188,
|
| 25 |
+
176,
|
| 26 |
+
211,
|
| 27 |
+
196,
|
| 28 |
+
193,
|
| 29 |
+
7,
|
| 30 |
+
187,
|
| 31 |
+
204,
|
| 32 |
+
185,
|
| 33 |
+
209,
|
| 34 |
+
182,
|
| 35 |
+
203,
|
| 36 |
+
201,
|
| 37 |
+
194,
|
| 38 |
+
0,
|
| 39 |
+
9,
|
| 40 |
+
212,
|
| 41 |
+
214,
|
| 42 |
+
183,
|
| 43 |
+
198,
|
| 44 |
+
191,
|
| 45 |
+
210,
|
| 46 |
+
93,
|
| 47 |
+
206,
|
| 48 |
+
208,
|
| 49 |
+
179,
|
| 50 |
+
178,
|
| 51 |
+
8,
|
| 52 |
+
205,
|
| 53 |
+
215,
|
| 54 |
+
6,
|
| 55 |
+
180,
|
| 56 |
+
195,
|
| 57 |
+
207,
|
| 58 |
+
199,
|
| 59 |
+
181,
|
| 60 |
+
197
|
| 61 |
+
],
|
| 62 |
+
"val_episodes": [
|
| 63 |
+
184,
|
| 64 |
+
189,
|
| 65 |
+
202,
|
| 66 |
+
4,
|
| 67 |
+
2
|
| 68 |
+
],
|
| 69 |
+
"train_layout_seeds": [
|
| 70 |
+
1339647934,
|
| 71 |
+
1966217163,
|
| 72 |
+
63402273,
|
| 73 |
+
4229107578,
|
| 74 |
+
3711669085,
|
| 75 |
+
3164807602,
|
| 76 |
+
975063281,
|
| 77 |
+
2833090125,
|
| 78 |
+
1248482997,
|
| 79 |
+
483767641,
|
| 80 |
+
2872056780,
|
| 81 |
+
3294494316,
|
| 82 |
+
961776701,
|
| 83 |
+
2700352190,
|
| 84 |
+
2296490322,
|
| 85 |
+
2554365269,
|
| 86 |
+
938316158,
|
| 87 |
+
3803324505,
|
| 88 |
+
2108098946,
|
| 89 |
+
2897428652,
|
| 90 |
+
1400654761,
|
| 91 |
+
2357694462,
|
| 92 |
+
1882561062,
|
| 93 |
+
2652290052,
|
| 94 |
+
2203817822,
|
| 95 |
+
4163076506,
|
| 96 |
+
1803864796,
|
| 97 |
+
3431188143,
|
| 98 |
+
2886057373,
|
| 99 |
+
4144272352,
|
| 100 |
+
465251528,
|
| 101 |
+
2637717298,
|
| 102 |
+
2653863089,
|
| 103 |
+
310325858,
|
| 104 |
+
3993727448,
|
| 105 |
+
194628840,
|
| 106 |
+
3265926662,
|
| 107 |
+
1275003304,
|
| 108 |
+
3213597487,
|
| 109 |
+
3441154191,
|
| 110 |
+
588342664,
|
| 111 |
+
2315029430,
|
| 112 |
+
3673039018,
|
| 113 |
+
2107735989,
|
| 114 |
+
3455910595
|
| 115 |
+
],
|
| 116 |
+
"val_layout_seeds": [
|
| 117 |
+
3206227309,
|
| 118 |
+
976180609,
|
| 119 |
+
1879278080,
|
| 120 |
+
4267076686,
|
| 121 |
+
2421463517
|
| 122 |
+
]
|
| 123 |
+
}
|
delta_pose/dataset_stats.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:87c63d4576e312d9c230da4f3f99c1c4998aa92810fc9cf6e749efe2643d9125
|
| 3 |
+
size 403
|
delta_pose/policy_best.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1b902921aac55cf5383922786c579513e11c3f399eda77789da01a0782bc6759
|
| 3 |
+
size 335850503
|