Commit
•
886d6d1
1
Parent(s):
22ce6d6
Upload 4 files
Browse files- .gitattributes +1 -0
- 300000/default/checkpoint +3 -0
- config.json +1887 -0
- dataset_statistics.json +1 -0
- example_batch.msgpack +3 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
300000/default/checkpoint filter=lfs diff=lfs merge=lfs -text
|
300000/default/checkpoint
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f1260979341c0c07ae4cf2480f924e71653c31b9c7b44ca8f64ae87bc1d61419
|
| 3 |
+
size 519021888
|
config.json
ADDED
|
@@ -0,0 +1,1887 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"dataset_kwargs": {
|
| 3 |
+
"balance_weights": false,
|
| 4 |
+
"batch_size": 8,
|
| 5 |
+
"dataset_kwargs_list": [
|
| 6 |
+
{
|
| 7 |
+
"name": "kuka",
|
| 8 |
+
"data_dir": null,
|
| 9 |
+
"image_obs_keys": {
|
| 10 |
+
"primary": "image",
|
| 11 |
+
"high": null,
|
| 12 |
+
"nav": null,
|
| 13 |
+
"left_wrist": null,
|
| 14 |
+
"right_wrist": null
|
| 15 |
+
},
|
| 16 |
+
"proprio_obs_keys": {
|
| 17 |
+
"bimanual": null,
|
| 18 |
+
"quadruped": null
|
| 19 |
+
},
|
| 20 |
+
"proprio_obs_dims": {
|
| 21 |
+
"bimanual": 14,
|
| 22 |
+
"quadruped": 59
|
| 23 |
+
},
|
| 24 |
+
"action_normalization_mask": [
|
| 25 |
+
true,
|
| 26 |
+
true,
|
| 27 |
+
true,
|
| 28 |
+
true,
|
| 29 |
+
true,
|
| 30 |
+
true,
|
| 31 |
+
false
|
| 32 |
+
],
|
| 33 |
+
"language_key": "language_instruction",
|
| 34 |
+
"action_proprio_normalization_type": "normal",
|
| 35 |
+
"standardize_fn": {
|
| 36 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 37 |
+
"name": "kuka_dataset_transform",
|
| 38 |
+
"args": [],
|
| 39 |
+
"kwargs": {}
|
| 40 |
+
}
|
| 41 |
+
},
|
| 42 |
+
{
|
| 43 |
+
"name": "taco_play",
|
| 44 |
+
"data_dir": null,
|
| 45 |
+
"image_obs_keys": {
|
| 46 |
+
"primary": "rgb_static",
|
| 47 |
+
"high": null,
|
| 48 |
+
"nav": null,
|
| 49 |
+
"left_wrist": null,
|
| 50 |
+
"right_wrist": null
|
| 51 |
+
},
|
| 52 |
+
"proprio_obs_keys": {
|
| 53 |
+
"bimanual": null,
|
| 54 |
+
"quadruped": null
|
| 55 |
+
},
|
| 56 |
+
"proprio_obs_dims": {
|
| 57 |
+
"bimanual": 14,
|
| 58 |
+
"quadruped": 59
|
| 59 |
+
},
|
| 60 |
+
"action_normalization_mask": [
|
| 61 |
+
true,
|
| 62 |
+
true,
|
| 63 |
+
true,
|
| 64 |
+
true,
|
| 65 |
+
true,
|
| 66 |
+
true,
|
| 67 |
+
false
|
| 68 |
+
],
|
| 69 |
+
"language_key": "language_instruction",
|
| 70 |
+
"action_proprio_normalization_type": "normal",
|
| 71 |
+
"standardize_fn": {
|
| 72 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 73 |
+
"name": "taco_dataset_transform",
|
| 74 |
+
"args": [],
|
| 75 |
+
"kwargs": {}
|
| 76 |
+
}
|
| 77 |
+
},
|
| 78 |
+
{
|
| 79 |
+
"name": "taco_extra",
|
| 80 |
+
"data_dir": null,
|
| 81 |
+
"image_obs_keys": {
|
| 82 |
+
"primary": "rgb_static",
|
| 83 |
+
"high": null,
|
| 84 |
+
"nav": null,
|
| 85 |
+
"left_wrist": null,
|
| 86 |
+
"right_wrist": null
|
| 87 |
+
},
|
| 88 |
+
"proprio_obs_keys": {
|
| 89 |
+
"bimanual": null,
|
| 90 |
+
"quadruped": null
|
| 91 |
+
},
|
| 92 |
+
"proprio_obs_dims": {
|
| 93 |
+
"bimanual": 14,
|
| 94 |
+
"quadruped": 59
|
| 95 |
+
},
|
| 96 |
+
"action_normalization_mask": [
|
| 97 |
+
true,
|
| 98 |
+
true,
|
| 99 |
+
true,
|
| 100 |
+
true,
|
| 101 |
+
true,
|
| 102 |
+
true,
|
| 103 |
+
false
|
| 104 |
+
],
|
| 105 |
+
"language_key": "language_instruction",
|
| 106 |
+
"action_proprio_normalization_type": "normal",
|
| 107 |
+
"standardize_fn": {
|
| 108 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 109 |
+
"name": "taco_dataset_transform",
|
| 110 |
+
"args": [],
|
| 111 |
+
"kwargs": {}
|
| 112 |
+
}
|
| 113 |
+
},
|
| 114 |
+
{
|
| 115 |
+
"name": "jaco_play",
|
| 116 |
+
"data_dir": null,
|
| 117 |
+
"image_obs_keys": {
|
| 118 |
+
"primary": "image",
|
| 119 |
+
"high": null,
|
| 120 |
+
"nav": null,
|
| 121 |
+
"left_wrist": null,
|
| 122 |
+
"right_wrist": null
|
| 123 |
+
},
|
| 124 |
+
"proprio_obs_keys": {
|
| 125 |
+
"bimanual": null,
|
| 126 |
+
"quadruped": null
|
| 127 |
+
},
|
| 128 |
+
"proprio_obs_dims": {
|
| 129 |
+
"bimanual": 14,
|
| 130 |
+
"quadruped": 59
|
| 131 |
+
},
|
| 132 |
+
"action_normalization_mask": [
|
| 133 |
+
true,
|
| 134 |
+
true,
|
| 135 |
+
true,
|
| 136 |
+
true,
|
| 137 |
+
true,
|
| 138 |
+
true,
|
| 139 |
+
false
|
| 140 |
+
],
|
| 141 |
+
"language_key": "language_instruction",
|
| 142 |
+
"action_proprio_normalization_type": "normal",
|
| 143 |
+
"standardize_fn": {
|
| 144 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 145 |
+
"name": "jaco_play_dataset_transform",
|
| 146 |
+
"args": [],
|
| 147 |
+
"kwargs": {}
|
| 148 |
+
}
|
| 149 |
+
},
|
| 150 |
+
{
|
| 151 |
+
"name": "berkeley_cable_routing",
|
| 152 |
+
"data_dir": null,
|
| 153 |
+
"image_obs_keys": {
|
| 154 |
+
"primary": "image",
|
| 155 |
+
"high": null,
|
| 156 |
+
"nav": null,
|
| 157 |
+
"left_wrist": null,
|
| 158 |
+
"right_wrist": null
|
| 159 |
+
},
|
| 160 |
+
"proprio_obs_keys": {
|
| 161 |
+
"bimanual": null,
|
| 162 |
+
"quadruped": null
|
| 163 |
+
},
|
| 164 |
+
"proprio_obs_dims": {
|
| 165 |
+
"bimanual": 14,
|
| 166 |
+
"quadruped": 59
|
| 167 |
+
},
|
| 168 |
+
"action_normalization_mask": [
|
| 169 |
+
true,
|
| 170 |
+
true,
|
| 171 |
+
true,
|
| 172 |
+
true,
|
| 173 |
+
true,
|
| 174 |
+
true,
|
| 175 |
+
false
|
| 176 |
+
],
|
| 177 |
+
"language_key": "language_instruction",
|
| 178 |
+
"action_proprio_normalization_type": "normal",
|
| 179 |
+
"standardize_fn": {
|
| 180 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 181 |
+
"name": "berkeley_cable_routing_dataset_transform",
|
| 182 |
+
"args": [],
|
| 183 |
+
"kwargs": {}
|
| 184 |
+
}
|
| 185 |
+
},
|
| 186 |
+
{
|
| 187 |
+
"name": "roboturk",
|
| 188 |
+
"data_dir": null,
|
| 189 |
+
"image_obs_keys": {
|
| 190 |
+
"primary": "front_rgb",
|
| 191 |
+
"high": null,
|
| 192 |
+
"nav": null,
|
| 193 |
+
"left_wrist": null,
|
| 194 |
+
"right_wrist": null
|
| 195 |
+
},
|
| 196 |
+
"proprio_obs_keys": {
|
| 197 |
+
"bimanual": null,
|
| 198 |
+
"quadruped": null
|
| 199 |
+
},
|
| 200 |
+
"proprio_obs_dims": {
|
| 201 |
+
"bimanual": 14,
|
| 202 |
+
"quadruped": 59
|
| 203 |
+
},
|
| 204 |
+
"action_normalization_mask": [
|
| 205 |
+
true,
|
| 206 |
+
true,
|
| 207 |
+
true,
|
| 208 |
+
true,
|
| 209 |
+
true,
|
| 210 |
+
true,
|
| 211 |
+
false
|
| 212 |
+
],
|
| 213 |
+
"language_key": "language_instruction",
|
| 214 |
+
"action_proprio_normalization_type": "normal",
|
| 215 |
+
"standardize_fn": {
|
| 216 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 217 |
+
"name": "roboturk_dataset_transform",
|
| 218 |
+
"args": [],
|
| 219 |
+
"kwargs": {}
|
| 220 |
+
}
|
| 221 |
+
},
|
| 222 |
+
{
|
| 223 |
+
"name": "viola",
|
| 224 |
+
"data_dir": null,
|
| 225 |
+
"image_obs_keys": {
|
| 226 |
+
"primary": "agentview_rgb",
|
| 227 |
+
"high": null,
|
| 228 |
+
"nav": null,
|
| 229 |
+
"left_wrist": null,
|
| 230 |
+
"right_wrist": null
|
| 231 |
+
},
|
| 232 |
+
"proprio_obs_keys": {
|
| 233 |
+
"bimanual": null,
|
| 234 |
+
"quadruped": null
|
| 235 |
+
},
|
| 236 |
+
"proprio_obs_dims": {
|
| 237 |
+
"bimanual": 14,
|
| 238 |
+
"quadruped": 59
|
| 239 |
+
},
|
| 240 |
+
"action_normalization_mask": [
|
| 241 |
+
true,
|
| 242 |
+
true,
|
| 243 |
+
true,
|
| 244 |
+
true,
|
| 245 |
+
true,
|
| 246 |
+
true,
|
| 247 |
+
false
|
| 248 |
+
],
|
| 249 |
+
"language_key": "language_instruction",
|
| 250 |
+
"action_proprio_normalization_type": "normal",
|
| 251 |
+
"standardize_fn": {
|
| 252 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 253 |
+
"name": "viola_dataset_transform",
|
| 254 |
+
"args": [],
|
| 255 |
+
"kwargs": {}
|
| 256 |
+
}
|
| 257 |
+
},
|
| 258 |
+
{
|
| 259 |
+
"name": "berkeley_autolab_ur5",
|
| 260 |
+
"data_dir": null,
|
| 261 |
+
"image_obs_keys": {
|
| 262 |
+
"primary": "image",
|
| 263 |
+
"high": null,
|
| 264 |
+
"nav": null,
|
| 265 |
+
"left_wrist": null,
|
| 266 |
+
"right_wrist": null
|
| 267 |
+
},
|
| 268 |
+
"proprio_obs_keys": {
|
| 269 |
+
"bimanual": null,
|
| 270 |
+
"quadruped": null
|
| 271 |
+
},
|
| 272 |
+
"proprio_obs_dims": {
|
| 273 |
+
"bimanual": 14,
|
| 274 |
+
"quadruped": 59
|
| 275 |
+
},
|
| 276 |
+
"action_normalization_mask": [
|
| 277 |
+
true,
|
| 278 |
+
true,
|
| 279 |
+
true,
|
| 280 |
+
true,
|
| 281 |
+
true,
|
| 282 |
+
true,
|
| 283 |
+
false
|
| 284 |
+
],
|
| 285 |
+
"language_key": "language_instruction",
|
| 286 |
+
"action_proprio_normalization_type": "normal",
|
| 287 |
+
"standardize_fn": {
|
| 288 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 289 |
+
"name": "berkeley_autolab_ur5_dataset_transform",
|
| 290 |
+
"args": [],
|
| 291 |
+
"kwargs": {}
|
| 292 |
+
}
|
| 293 |
+
},
|
| 294 |
+
{
|
| 295 |
+
"name": "toto",
|
| 296 |
+
"data_dir": null,
|
| 297 |
+
"image_obs_keys": {
|
| 298 |
+
"primary": "image",
|
| 299 |
+
"high": null,
|
| 300 |
+
"nav": null,
|
| 301 |
+
"left_wrist": null,
|
| 302 |
+
"right_wrist": null
|
| 303 |
+
},
|
| 304 |
+
"proprio_obs_keys": {
|
| 305 |
+
"bimanual": null,
|
| 306 |
+
"quadruped": null
|
| 307 |
+
},
|
| 308 |
+
"proprio_obs_dims": {
|
| 309 |
+
"bimanual": 14,
|
| 310 |
+
"quadruped": 59
|
| 311 |
+
},
|
| 312 |
+
"action_normalization_mask": [
|
| 313 |
+
true,
|
| 314 |
+
true,
|
| 315 |
+
true,
|
| 316 |
+
true,
|
| 317 |
+
true,
|
| 318 |
+
true,
|
| 319 |
+
false
|
| 320 |
+
],
|
| 321 |
+
"language_key": "language_instruction",
|
| 322 |
+
"action_proprio_normalization_type": "normal",
|
| 323 |
+
"standardize_fn": {
|
| 324 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 325 |
+
"name": "toto_dataset_transform",
|
| 326 |
+
"args": [],
|
| 327 |
+
"kwargs": {}
|
| 328 |
+
}
|
| 329 |
+
},
|
| 330 |
+
{
|
| 331 |
+
"name": "language_table",
|
| 332 |
+
"data_dir": null,
|
| 333 |
+
"image_obs_keys": {
|
| 334 |
+
"primary": "rgb",
|
| 335 |
+
"high": null,
|
| 336 |
+
"nav": null,
|
| 337 |
+
"left_wrist": null,
|
| 338 |
+
"right_wrist": null
|
| 339 |
+
},
|
| 340 |
+
"proprio_obs_keys": {
|
| 341 |
+
"bimanual": null,
|
| 342 |
+
"quadruped": null
|
| 343 |
+
},
|
| 344 |
+
"proprio_obs_dims": {
|
| 345 |
+
"bimanual": 14,
|
| 346 |
+
"quadruped": 59
|
| 347 |
+
},
|
| 348 |
+
"action_normalization_mask": [
|
| 349 |
+
true,
|
| 350 |
+
true,
|
| 351 |
+
true,
|
| 352 |
+
true,
|
| 353 |
+
true,
|
| 354 |
+
true,
|
| 355 |
+
false
|
| 356 |
+
],
|
| 357 |
+
"language_key": "language_instruction",
|
| 358 |
+
"action_proprio_normalization_type": "normal",
|
| 359 |
+
"standardize_fn": {
|
| 360 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 361 |
+
"name": "language_table_dataset_transform",
|
| 362 |
+
"args": [],
|
| 363 |
+
"kwargs": {}
|
| 364 |
+
}
|
| 365 |
+
},
|
| 366 |
+
{
|
| 367 |
+
"name": "stanford_hydra_dataset_converted_externally_to_rlds",
|
| 368 |
+
"data_dir": null,
|
| 369 |
+
"image_obs_keys": {
|
| 370 |
+
"primary": "image",
|
| 371 |
+
"high": null,
|
| 372 |
+
"nav": null,
|
| 373 |
+
"left_wrist": null,
|
| 374 |
+
"right_wrist": null
|
| 375 |
+
},
|
| 376 |
+
"proprio_obs_keys": {
|
| 377 |
+
"bimanual": null,
|
| 378 |
+
"quadruped": null
|
| 379 |
+
},
|
| 380 |
+
"proprio_obs_dims": {
|
| 381 |
+
"bimanual": 14,
|
| 382 |
+
"quadruped": 59
|
| 383 |
+
},
|
| 384 |
+
"action_normalization_mask": [
|
| 385 |
+
true,
|
| 386 |
+
true,
|
| 387 |
+
true,
|
| 388 |
+
true,
|
| 389 |
+
true,
|
| 390 |
+
true,
|
| 391 |
+
false
|
| 392 |
+
],
|
| 393 |
+
"language_key": "language_instruction",
|
| 394 |
+
"action_proprio_normalization_type": "normal",
|
| 395 |
+
"standardize_fn": {
|
| 396 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 397 |
+
"name": "stanford_hydra_dataset_transform",
|
| 398 |
+
"args": [],
|
| 399 |
+
"kwargs": {}
|
| 400 |
+
}
|
| 401 |
+
},
|
| 402 |
+
{
|
| 403 |
+
"name": "austin_buds_dataset_converted_externally_to_rlds",
|
| 404 |
+
"data_dir": null,
|
| 405 |
+
"image_obs_keys": {
|
| 406 |
+
"primary": "image",
|
| 407 |
+
"high": null,
|
| 408 |
+
"nav": null,
|
| 409 |
+
"left_wrist": null,
|
| 410 |
+
"right_wrist": null
|
| 411 |
+
},
|
| 412 |
+
"proprio_obs_keys": {
|
| 413 |
+
"bimanual": null,
|
| 414 |
+
"quadruped": null
|
| 415 |
+
},
|
| 416 |
+
"proprio_obs_dims": {
|
| 417 |
+
"bimanual": 14,
|
| 418 |
+
"quadruped": 59
|
| 419 |
+
},
|
| 420 |
+
"action_normalization_mask": [
|
| 421 |
+
true,
|
| 422 |
+
true,
|
| 423 |
+
true,
|
| 424 |
+
true,
|
| 425 |
+
true,
|
| 426 |
+
true,
|
| 427 |
+
false
|
| 428 |
+
],
|
| 429 |
+
"language_key": "language_instruction",
|
| 430 |
+
"action_proprio_normalization_type": "normal",
|
| 431 |
+
"standardize_fn": {
|
| 432 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 433 |
+
"name": "austin_buds_dataset_transform",
|
| 434 |
+
"args": [],
|
| 435 |
+
"kwargs": {}
|
| 436 |
+
}
|
| 437 |
+
},
|
| 438 |
+
{
|
| 439 |
+
"name": "nyu_franka_play_dataset_converted_externally_to_rlds",
|
| 440 |
+
"data_dir": null,
|
| 441 |
+
"image_obs_keys": {
|
| 442 |
+
"primary": "image",
|
| 443 |
+
"high": null,
|
| 444 |
+
"nav": null,
|
| 445 |
+
"left_wrist": null,
|
| 446 |
+
"right_wrist": null
|
| 447 |
+
},
|
| 448 |
+
"proprio_obs_keys": {
|
| 449 |
+
"bimanual": null,
|
| 450 |
+
"quadruped": null
|
| 451 |
+
},
|
| 452 |
+
"proprio_obs_dims": {
|
| 453 |
+
"bimanual": 14,
|
| 454 |
+
"quadruped": 59
|
| 455 |
+
},
|
| 456 |
+
"action_normalization_mask": [
|
| 457 |
+
true,
|
| 458 |
+
true,
|
| 459 |
+
true,
|
| 460 |
+
true,
|
| 461 |
+
true,
|
| 462 |
+
true,
|
| 463 |
+
false
|
| 464 |
+
],
|
| 465 |
+
"language_key": "language_instruction",
|
| 466 |
+
"action_proprio_normalization_type": "normal",
|
| 467 |
+
"standardize_fn": {
|
| 468 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 469 |
+
"name": "nyu_franka_play_dataset_transform",
|
| 470 |
+
"args": [],
|
| 471 |
+
"kwargs": {}
|
| 472 |
+
}
|
| 473 |
+
},
|
| 474 |
+
{
|
| 475 |
+
"name": "furniture_bench_dataset_converted_externally_to_rlds",
|
| 476 |
+
"data_dir": null,
|
| 477 |
+
"image_obs_keys": {
|
| 478 |
+
"primary": "image",
|
| 479 |
+
"high": null,
|
| 480 |
+
"nav": null,
|
| 481 |
+
"left_wrist": null,
|
| 482 |
+
"right_wrist": null
|
| 483 |
+
},
|
| 484 |
+
"proprio_obs_keys": {
|
| 485 |
+
"bimanual": null,
|
| 486 |
+
"quadruped": null
|
| 487 |
+
},
|
| 488 |
+
"proprio_obs_dims": {
|
| 489 |
+
"bimanual": 14,
|
| 490 |
+
"quadruped": 59
|
| 491 |
+
},
|
| 492 |
+
"action_normalization_mask": [
|
| 493 |
+
true,
|
| 494 |
+
true,
|
| 495 |
+
true,
|
| 496 |
+
true,
|
| 497 |
+
true,
|
| 498 |
+
true,
|
| 499 |
+
false
|
| 500 |
+
],
|
| 501 |
+
"language_key": "language_instruction",
|
| 502 |
+
"action_proprio_normalization_type": "normal",
|
| 503 |
+
"standardize_fn": {
|
| 504 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 505 |
+
"name": "furniture_bench_dataset_transform",
|
| 506 |
+
"args": [],
|
| 507 |
+
"kwargs": {}
|
| 508 |
+
}
|
| 509 |
+
},
|
| 510 |
+
{
|
| 511 |
+
"name": "austin_sailor_dataset_converted_externally_to_rlds",
|
| 512 |
+
"data_dir": null,
|
| 513 |
+
"image_obs_keys": {
|
| 514 |
+
"primary": "image",
|
| 515 |
+
"nav": null,
|
| 516 |
+
"high": null,
|
| 517 |
+
"left_wrist": null,
|
| 518 |
+
"right_wrist": null
|
| 519 |
+
},
|
| 520 |
+
"proprio_obs_keys": {
|
| 521 |
+
"bimanual": null,
|
| 522 |
+
"quadruped": null
|
| 523 |
+
},
|
| 524 |
+
"proprio_obs_dims": {
|
| 525 |
+
"bimanual": 14,
|
| 526 |
+
"quadruped": 59
|
| 527 |
+
},
|
| 528 |
+
"action_normalization_mask": [
|
| 529 |
+
true,
|
| 530 |
+
true,
|
| 531 |
+
true,
|
| 532 |
+
true,
|
| 533 |
+
true,
|
| 534 |
+
true,
|
| 535 |
+
false
|
| 536 |
+
],
|
| 537 |
+
"language_key": "language_instruction",
|
| 538 |
+
"action_proprio_normalization_type": "normal",
|
| 539 |
+
"standardize_fn": {
|
| 540 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 541 |
+
"name": "austin_sailor_dataset_transform",
|
| 542 |
+
"args": [],
|
| 543 |
+
"kwargs": {}
|
| 544 |
+
}
|
| 545 |
+
},
|
| 546 |
+
{
|
| 547 |
+
"name": "austin_sirius_dataset_converted_externally_to_rlds",
|
| 548 |
+
"data_dir": null,
|
| 549 |
+
"image_obs_keys": {
|
| 550 |
+
"primary": "image",
|
| 551 |
+
"nav": null,
|
| 552 |
+
"high": null,
|
| 553 |
+
"left_wrist": null,
|
| 554 |
+
"right_wrist": null
|
| 555 |
+
},
|
| 556 |
+
"proprio_obs_keys": {
|
| 557 |
+
"bimanual": null,
|
| 558 |
+
"quadruped": null
|
| 559 |
+
},
|
| 560 |
+
"proprio_obs_dims": {
|
| 561 |
+
"bimanual": 14,
|
| 562 |
+
"quadruped": 59
|
| 563 |
+
},
|
| 564 |
+
"action_normalization_mask": [
|
| 565 |
+
true,
|
| 566 |
+
true,
|
| 567 |
+
true,
|
| 568 |
+
true,
|
| 569 |
+
true,
|
| 570 |
+
true,
|
| 571 |
+
false
|
| 572 |
+
],
|
| 573 |
+
"language_key": "language_instruction",
|
| 574 |
+
"action_proprio_normalization_type": "normal",
|
| 575 |
+
"standardize_fn": {
|
| 576 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 577 |
+
"name": "austin_sirius_dataset_transform",
|
| 578 |
+
"args": [],
|
| 579 |
+
"kwargs": {}
|
| 580 |
+
}
|
| 581 |
+
},
|
| 582 |
+
{
|
| 583 |
+
"name": "bc_z",
|
| 584 |
+
"data_dir": null,
|
| 585 |
+
"image_obs_keys": {
|
| 586 |
+
"primary": "image",
|
| 587 |
+
"high": null,
|
| 588 |
+
"nav": null,
|
| 589 |
+
"left_wrist": null,
|
| 590 |
+
"right_wrist": null
|
| 591 |
+
},
|
| 592 |
+
"proprio_obs_keys": {
|
| 593 |
+
"bimanual": null,
|
| 594 |
+
"quadruped": null
|
| 595 |
+
},
|
| 596 |
+
"proprio_obs_dims": {
|
| 597 |
+
"bimanual": 14,
|
| 598 |
+
"quadruped": 59
|
| 599 |
+
},
|
| 600 |
+
"action_normalization_mask": [
|
| 601 |
+
true,
|
| 602 |
+
true,
|
| 603 |
+
true,
|
| 604 |
+
true,
|
| 605 |
+
true,
|
| 606 |
+
true,
|
| 607 |
+
false
|
| 608 |
+
],
|
| 609 |
+
"language_key": "language_instruction",
|
| 610 |
+
"action_proprio_normalization_type": "normal",
|
| 611 |
+
"standardize_fn": {
|
| 612 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 613 |
+
"name": "bc_z_dataset_transform",
|
| 614 |
+
"args": [],
|
| 615 |
+
"kwargs": {}
|
| 616 |
+
}
|
| 617 |
+
},
|
| 618 |
+
{
|
| 619 |
+
"name": "dlr_edan_shared_control_converted_externally_to_rlds",
|
| 620 |
+
"data_dir": null,
|
| 621 |
+
"image_obs_keys": {
|
| 622 |
+
"primary": "image",
|
| 623 |
+
"high": null,
|
| 624 |
+
"nav": null,
|
| 625 |
+
"left_wrist": null,
|
| 626 |
+
"right_wrist": null
|
| 627 |
+
},
|
| 628 |
+
"proprio_obs_keys": {
|
| 629 |
+
"bimanual": null,
|
| 630 |
+
"quadruped": null
|
| 631 |
+
},
|
| 632 |
+
"proprio_obs_dims": {
|
| 633 |
+
"bimanual": 14,
|
| 634 |
+
"quadruped": 59
|
| 635 |
+
},
|
| 636 |
+
"action_normalization_mask": [
|
| 637 |
+
true,
|
| 638 |
+
true,
|
| 639 |
+
true,
|
| 640 |
+
true,
|
| 641 |
+
true,
|
| 642 |
+
true,
|
| 643 |
+
false
|
| 644 |
+
],
|
| 645 |
+
"language_key": "language_instruction",
|
| 646 |
+
"action_proprio_normalization_type": "normal",
|
| 647 |
+
"standardize_fn": {
|
| 648 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 649 |
+
"name": "dlr_edan_shared_control_dataset_transform",
|
| 650 |
+
"args": [],
|
| 651 |
+
"kwargs": {}
|
| 652 |
+
}
|
| 653 |
+
},
|
| 654 |
+
{
|
| 655 |
+
"name": "iamlab_cmu_pickup_insert_converted_externally_to_rlds",
|
| 656 |
+
"data_dir": null,
|
| 657 |
+
"image_obs_keys": {
|
| 658 |
+
"primary": "image",
|
| 659 |
+
"high": null,
|
| 660 |
+
"nav": null,
|
| 661 |
+
"left_wrist": null,
|
| 662 |
+
"right_wrist": null
|
| 663 |
+
},
|
| 664 |
+
"proprio_obs_keys": {
|
| 665 |
+
"bimanual": null,
|
| 666 |
+
"quadruped": null
|
| 667 |
+
},
|
| 668 |
+
"proprio_obs_dims": {
|
| 669 |
+
"bimanual": 14,
|
| 670 |
+
"quadruped": 59
|
| 671 |
+
},
|
| 672 |
+
"action_normalization_mask": [
|
| 673 |
+
true,
|
| 674 |
+
true,
|
| 675 |
+
true,
|
| 676 |
+
true,
|
| 677 |
+
true,
|
| 678 |
+
true,
|
| 679 |
+
false
|
| 680 |
+
],
|
| 681 |
+
"language_key": "language_instruction",
|
| 682 |
+
"action_proprio_normalization_type": "normal",
|
| 683 |
+
"standardize_fn": {
|
| 684 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 685 |
+
"name": "iamlab_pick_insert_dataset_transform",
|
| 686 |
+
"args": [],
|
| 687 |
+
"kwargs": {}
|
| 688 |
+
}
|
| 689 |
+
},
|
| 690 |
+
{
|
| 691 |
+
"name": "utaustin_mutex",
|
| 692 |
+
"data_dir": null,
|
| 693 |
+
"image_obs_keys": {
|
| 694 |
+
"primary": "image",
|
| 695 |
+
"high": null,
|
| 696 |
+
"nav": null,
|
| 697 |
+
"left_wrist": null,
|
| 698 |
+
"right_wrist": null
|
| 699 |
+
},
|
| 700 |
+
"proprio_obs_keys": {
|
| 701 |
+
"bimanual": null,
|
| 702 |
+
"quadruped": null
|
| 703 |
+
},
|
| 704 |
+
"proprio_obs_dims": {
|
| 705 |
+
"bimanual": 14,
|
| 706 |
+
"quadruped": 59
|
| 707 |
+
},
|
| 708 |
+
"action_normalization_mask": [
|
| 709 |
+
true,
|
| 710 |
+
true,
|
| 711 |
+
true,
|
| 712 |
+
true,
|
| 713 |
+
true,
|
| 714 |
+
true,
|
| 715 |
+
false
|
| 716 |
+
],
|
| 717 |
+
"language_key": "language_instruction",
|
| 718 |
+
"action_proprio_normalization_type": "normal",
|
| 719 |
+
"standardize_fn": {
|
| 720 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 721 |
+
"name": "utaustin_mutex_dataset_transform",
|
| 722 |
+
"args": [],
|
| 723 |
+
"kwargs": {}
|
| 724 |
+
}
|
| 725 |
+
},
|
| 726 |
+
{
|
| 727 |
+
"name": "berkeley_fanuc_manipulation",
|
| 728 |
+
"data_dir": null,
|
| 729 |
+
"image_obs_keys": {
|
| 730 |
+
"primary": "image",
|
| 731 |
+
"high": null,
|
| 732 |
+
"nav": null,
|
| 733 |
+
"left_wrist": null,
|
| 734 |
+
"right_wrist": null
|
| 735 |
+
},
|
| 736 |
+
"proprio_obs_keys": {
|
| 737 |
+
"bimanual": null,
|
| 738 |
+
"quadruped": null
|
| 739 |
+
},
|
| 740 |
+
"proprio_obs_dims": {
|
| 741 |
+
"bimanual": 14,
|
| 742 |
+
"quadruped": 59
|
| 743 |
+
},
|
| 744 |
+
"action_normalization_mask": [
|
| 745 |
+
true,
|
| 746 |
+
true,
|
| 747 |
+
true,
|
| 748 |
+
true,
|
| 749 |
+
true,
|
| 750 |
+
true,
|
| 751 |
+
false
|
| 752 |
+
],
|
| 753 |
+
"language_key": "language_instruction",
|
| 754 |
+
"action_proprio_normalization_type": "normal",
|
| 755 |
+
"standardize_fn": {
|
| 756 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 757 |
+
"name": "berkeley_fanuc_dataset_transform",
|
| 758 |
+
"args": [],
|
| 759 |
+
"kwargs": {}
|
| 760 |
+
}
|
| 761 |
+
},
|
| 762 |
+
{
|
| 763 |
+
"name": "cmu_stretch",
|
| 764 |
+
"data_dir": null,
|
| 765 |
+
"image_obs_keys": {
|
| 766 |
+
"primary": "image",
|
| 767 |
+
"high": null,
|
| 768 |
+
"nav": null,
|
| 769 |
+
"left_wrist": null,
|
| 770 |
+
"right_wrist": null
|
| 771 |
+
},
|
| 772 |
+
"proprio_obs_keys": {
|
| 773 |
+
"bimanual": null,
|
| 774 |
+
"quadruped": null
|
| 775 |
+
},
|
| 776 |
+
"proprio_obs_dims": {
|
| 777 |
+
"bimanual": 14,
|
| 778 |
+
"quadruped": 59
|
| 779 |
+
},
|
| 780 |
+
"action_normalization_mask": [
|
| 781 |
+
true,
|
| 782 |
+
true,
|
| 783 |
+
true,
|
| 784 |
+
true,
|
| 785 |
+
true,
|
| 786 |
+
true,
|
| 787 |
+
false
|
| 788 |
+
],
|
| 789 |
+
"language_key": "language_instruction",
|
| 790 |
+
"action_proprio_normalization_type": "normal",
|
| 791 |
+
"standardize_fn": {
|
| 792 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 793 |
+
"name": "cmu_stretch_dataset_transform",
|
| 794 |
+
"args": [],
|
| 795 |
+
"kwargs": {}
|
| 796 |
+
}
|
| 797 |
+
},
|
| 798 |
+
{
|
| 799 |
+
"name": "droid",
|
| 800 |
+
"data_dir": null,
|
| 801 |
+
"image_obs_keys": {
|
| 802 |
+
"primary": "exterior_image_1_left",
|
| 803 |
+
"high": null,
|
| 804 |
+
"nav": null,
|
| 805 |
+
"left_wrist": null,
|
| 806 |
+
"right_wrist": null
|
| 807 |
+
},
|
| 808 |
+
"proprio_obs_keys": {
|
| 809 |
+
"bimanual": null,
|
| 810 |
+
"quadruped": null
|
| 811 |
+
},
|
| 812 |
+
"proprio_obs_dims": {
|
| 813 |
+
"bimanual": 14,
|
| 814 |
+
"quadruped": 59
|
| 815 |
+
},
|
| 816 |
+
"filter_functions": [
|
| 817 |
+
{
|
| 818 |
+
"module": "experiments.homer.scripts.configs.cross_embodiment",
|
| 819 |
+
"name": "filter_success",
|
| 820 |
+
"args": [],
|
| 821 |
+
"kwargs": {}
|
| 822 |
+
}
|
| 823 |
+
],
|
| 824 |
+
"action_normalization_mask": [
|
| 825 |
+
true,
|
| 826 |
+
true,
|
| 827 |
+
true,
|
| 828 |
+
true,
|
| 829 |
+
true,
|
| 830 |
+
true,
|
| 831 |
+
false
|
| 832 |
+
],
|
| 833 |
+
"language_key": "language_instruction",
|
| 834 |
+
"action_proprio_normalization_type": "normal",
|
| 835 |
+
"standardize_fn": {
|
| 836 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 837 |
+
"name": "droid_dataset_transform",
|
| 838 |
+
"args": [],
|
| 839 |
+
"kwargs": {}
|
| 840 |
+
}
|
| 841 |
+
},
|
| 842 |
+
{
|
| 843 |
+
"name": "aloha_pen_uncap_diverse_dataset",
|
| 844 |
+
"data_dir": null,
|
| 845 |
+
"image_obs_keys": {
|
| 846 |
+
"primary": null,
|
| 847 |
+
"high": "cam_high",
|
| 848 |
+
"nav": null,
|
| 849 |
+
"left_wrist": "cam_left_wrist",
|
| 850 |
+
"right_wrist": "cam_right_wrist"
|
| 851 |
+
},
|
| 852 |
+
"proprio_obs_keys": {
|
| 853 |
+
"bimanual": "proprio",
|
| 854 |
+
"quadruped": null
|
| 855 |
+
},
|
| 856 |
+
"proprio_obs_dims": {
|
| 857 |
+
"bimanual": 14,
|
| 858 |
+
"quadruped": 59
|
| 859 |
+
},
|
| 860 |
+
"action_normalization_mask": [
|
| 861 |
+
true,
|
| 862 |
+
true,
|
| 863 |
+
true,
|
| 864 |
+
true,
|
| 865 |
+
true,
|
| 866 |
+
true,
|
| 867 |
+
false,
|
| 868 |
+
true,
|
| 869 |
+
true,
|
| 870 |
+
true,
|
| 871 |
+
true,
|
| 872 |
+
true,
|
| 873 |
+
true,
|
| 874 |
+
false
|
| 875 |
+
],
|
| 876 |
+
"language_key": "language_instruction",
|
| 877 |
+
"action_proprio_normalization_type": "normal",
|
| 878 |
+
"standardize_fn": {
|
| 879 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 880 |
+
"name": "aloha_pen_uncap_dataset_transform",
|
| 881 |
+
"args": [],
|
| 882 |
+
"kwargs": {}
|
| 883 |
+
}
|
| 884 |
+
},
|
| 885 |
+
{
|
| 886 |
+
"name": "aloha_dough_cut_dataset",
|
| 887 |
+
"data_dir": null,
|
| 888 |
+
"image_obs_keys": {
|
| 889 |
+
"primary": null,
|
| 890 |
+
"high": "cam_high",
|
| 891 |
+
"nav": null,
|
| 892 |
+
"left_wrist": "cam_left_wrist",
|
| 893 |
+
"right_wrist": "cam_right_wrist"
|
| 894 |
+
},
|
| 895 |
+
"proprio_obs_keys": {
|
| 896 |
+
"bimanual": "proprio",
|
| 897 |
+
"quadruped": null
|
| 898 |
+
},
|
| 899 |
+
"proprio_obs_dims": {
|
| 900 |
+
"bimanual": 14,
|
| 901 |
+
"quadruped": 59
|
| 902 |
+
},
|
| 903 |
+
"action_normalization_mask": [
|
| 904 |
+
true,
|
| 905 |
+
true,
|
| 906 |
+
true,
|
| 907 |
+
true,
|
| 908 |
+
true,
|
| 909 |
+
true,
|
| 910 |
+
false,
|
| 911 |
+
true,
|
| 912 |
+
true,
|
| 913 |
+
true,
|
| 914 |
+
true,
|
| 915 |
+
true,
|
| 916 |
+
true,
|
| 917 |
+
false
|
| 918 |
+
],
|
| 919 |
+
"language_key": "language_instruction",
|
| 920 |
+
"action_proprio_normalization_type": "normal",
|
| 921 |
+
"standardize_fn": {
|
| 922 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 923 |
+
"name": "aloha_dough_dataset_transform",
|
| 924 |
+
"args": [],
|
| 925 |
+
"kwargs": {}
|
| 926 |
+
}
|
| 927 |
+
},
|
| 928 |
+
{
|
| 929 |
+
"name": "aloha_lucy_dataset",
|
| 930 |
+
"data_dir": null,
|
| 931 |
+
"image_obs_keys": {
|
| 932 |
+
"primary": null,
|
| 933 |
+
"high": "cam_high",
|
| 934 |
+
"nav": null,
|
| 935 |
+
"left_wrist": "cam_left_wrist",
|
| 936 |
+
"right_wrist": "cam_right_wrist"
|
| 937 |
+
},
|
| 938 |
+
"proprio_obs_keys": {
|
| 939 |
+
"bimanual": "proprio",
|
| 940 |
+
"quadruped": null
|
| 941 |
+
},
|
| 942 |
+
"proprio_obs_dims": {
|
| 943 |
+
"bimanual": 14,
|
| 944 |
+
"quadruped": 59
|
| 945 |
+
},
|
| 946 |
+
"action_normalization_mask": [
|
| 947 |
+
true,
|
| 948 |
+
true,
|
| 949 |
+
true,
|
| 950 |
+
true,
|
| 951 |
+
true,
|
| 952 |
+
true,
|
| 953 |
+
false,
|
| 954 |
+
true,
|
| 955 |
+
true,
|
| 956 |
+
true,
|
| 957 |
+
true,
|
| 958 |
+
true,
|
| 959 |
+
true,
|
| 960 |
+
false
|
| 961 |
+
],
|
| 962 |
+
"language_key": "language_instruction",
|
| 963 |
+
"action_proprio_normalization_type": "normal",
|
| 964 |
+
"standardize_fn": {
|
| 965 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 966 |
+
"name": "aloha_dough_dataset_transform",
|
| 967 |
+
"args": [],
|
| 968 |
+
"kwargs": {}
|
| 969 |
+
}
|
| 970 |
+
},
|
| 971 |
+
{
|
| 972 |
+
"name": "aloha_drawer_dataset",
|
| 973 |
+
"data_dir": null,
|
| 974 |
+
"image_obs_keys": {
|
| 975 |
+
"primary": null,
|
| 976 |
+
"high": "cam_high",
|
| 977 |
+
"nav": null,
|
| 978 |
+
"left_wrist": "cam_left_wrist",
|
| 979 |
+
"right_wrist": "cam_right_wrist"
|
| 980 |
+
},
|
| 981 |
+
"proprio_obs_keys": {
|
| 982 |
+
"bimanual": "proprio",
|
| 983 |
+
"quadruped": null
|
| 984 |
+
},
|
| 985 |
+
"proprio_obs_dims": {
|
| 986 |
+
"bimanual": 14,
|
| 987 |
+
"quadruped": 59
|
| 988 |
+
},
|
| 989 |
+
"action_normalization_mask": [
|
| 990 |
+
true,
|
| 991 |
+
true,
|
| 992 |
+
true,
|
| 993 |
+
true,
|
| 994 |
+
true,
|
| 995 |
+
true,
|
| 996 |
+
false,
|
| 997 |
+
true,
|
| 998 |
+
true,
|
| 999 |
+
true,
|
| 1000 |
+
true,
|
| 1001 |
+
true,
|
| 1002 |
+
true,
|
| 1003 |
+
false
|
| 1004 |
+
],
|
| 1005 |
+
"language_key": "language_instruction",
|
| 1006 |
+
"action_proprio_normalization_type": "normal",
|
| 1007 |
+
"standardize_fn": {
|
| 1008 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 1009 |
+
"name": "aloha_dough_dataset_transform",
|
| 1010 |
+
"args": [],
|
| 1011 |
+
"kwargs": {}
|
| 1012 |
+
}
|
| 1013 |
+
},
|
| 1014 |
+
{
|
| 1015 |
+
"name": "aloha_pick_place_dataset",
|
| 1016 |
+
"data_dir": null,
|
| 1017 |
+
"image_obs_keys": {
|
| 1018 |
+
"primary": null,
|
| 1019 |
+
"high": "cam_high",
|
| 1020 |
+
"nav": null,
|
| 1021 |
+
"left_wrist": "cam_left_wrist",
|
| 1022 |
+
"right_wrist": "cam_right_wrist"
|
| 1023 |
+
},
|
| 1024 |
+
"proprio_obs_keys": {
|
| 1025 |
+
"bimanual": "proprio",
|
| 1026 |
+
"quadruped": null
|
| 1027 |
+
},
|
| 1028 |
+
"proprio_obs_dims": {
|
| 1029 |
+
"bimanual": 14,
|
| 1030 |
+
"quadruped": 59
|
| 1031 |
+
},
|
| 1032 |
+
"action_normalization_mask": [
|
| 1033 |
+
true,
|
| 1034 |
+
true,
|
| 1035 |
+
true,
|
| 1036 |
+
true,
|
| 1037 |
+
true,
|
| 1038 |
+
true,
|
| 1039 |
+
false,
|
| 1040 |
+
true,
|
| 1041 |
+
true,
|
| 1042 |
+
true,
|
| 1043 |
+
true,
|
| 1044 |
+
true,
|
| 1045 |
+
true,
|
| 1046 |
+
false
|
| 1047 |
+
],
|
| 1048 |
+
"language_key": "language_instruction",
|
| 1049 |
+
"action_proprio_normalization_type": "normal",
|
| 1050 |
+
"standardize_fn": {
|
| 1051 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 1052 |
+
"name": "aloha_dough_dataset_transform",
|
| 1053 |
+
"args": [],
|
| 1054 |
+
"kwargs": {}
|
| 1055 |
+
}
|
| 1056 |
+
},
|
| 1057 |
+
{
|
| 1058 |
+
"name": "aloha_static_dataset",
|
| 1059 |
+
"data_dir": null,
|
| 1060 |
+
"image_obs_keys": {
|
| 1061 |
+
"primary": null,
|
| 1062 |
+
"high": "cam_high",
|
| 1063 |
+
"nav": null,
|
| 1064 |
+
"left_wrist": "cam_left_wrist",
|
| 1065 |
+
"right_wrist": "cam_right_wrist"
|
| 1066 |
+
},
|
| 1067 |
+
"proprio_obs_keys": {
|
| 1068 |
+
"bimanual": "proprio",
|
| 1069 |
+
"quadruped": null
|
| 1070 |
+
},
|
| 1071 |
+
"proprio_obs_dims": {
|
| 1072 |
+
"bimanual": 14,
|
| 1073 |
+
"quadruped": 59
|
| 1074 |
+
},
|
| 1075 |
+
"action_normalization_mask": [
|
| 1076 |
+
true,
|
| 1077 |
+
true,
|
| 1078 |
+
true,
|
| 1079 |
+
true,
|
| 1080 |
+
true,
|
| 1081 |
+
true,
|
| 1082 |
+
false,
|
| 1083 |
+
true,
|
| 1084 |
+
true,
|
| 1085 |
+
true,
|
| 1086 |
+
true,
|
| 1087 |
+
true,
|
| 1088 |
+
true,
|
| 1089 |
+
false
|
| 1090 |
+
],
|
| 1091 |
+
"language_key": "language_instruction",
|
| 1092 |
+
"action_proprio_normalization_type": "normal",
|
| 1093 |
+
"standardize_fn": {
|
| 1094 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 1095 |
+
"name": "aloha_dough_dataset_transform",
|
| 1096 |
+
"args": [],
|
| 1097 |
+
"kwargs": {}
|
| 1098 |
+
}
|
| 1099 |
+
},
|
| 1100 |
+
{
|
| 1101 |
+
"name": "aloha_sushi_cut_full_dataset",
|
| 1102 |
+
"data_dir": null,
|
| 1103 |
+
"image_obs_keys": {
|
| 1104 |
+
"primary": null,
|
| 1105 |
+
"high": "cam_high",
|
| 1106 |
+
"nav": null,
|
| 1107 |
+
"left_wrist": "cam_left_wrist",
|
| 1108 |
+
"right_wrist": "cam_right_wrist"
|
| 1109 |
+
},
|
| 1110 |
+
"proprio_obs_keys": {
|
| 1111 |
+
"bimanual": "proprio",
|
| 1112 |
+
"quadruped": null
|
| 1113 |
+
},
|
| 1114 |
+
"proprio_obs_dims": {
|
| 1115 |
+
"bimanual": 14,
|
| 1116 |
+
"quadruped": 59
|
| 1117 |
+
},
|
| 1118 |
+
"action_normalization_mask": [
|
| 1119 |
+
true,
|
| 1120 |
+
true,
|
| 1121 |
+
true,
|
| 1122 |
+
true,
|
| 1123 |
+
true,
|
| 1124 |
+
true,
|
| 1125 |
+
false,
|
| 1126 |
+
true,
|
| 1127 |
+
true,
|
| 1128 |
+
true,
|
| 1129 |
+
true,
|
| 1130 |
+
true,
|
| 1131 |
+
true,
|
| 1132 |
+
false
|
| 1133 |
+
],
|
| 1134 |
+
"language_key": "language_instruction",
|
| 1135 |
+
"action_proprio_normalization_type": "normal",
|
| 1136 |
+
"standardize_fn": {
|
| 1137 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 1138 |
+
"name": "aloha_dough_dataset_transform",
|
| 1139 |
+
"args": [],
|
| 1140 |
+
"kwargs": {}
|
| 1141 |
+
}
|
| 1142 |
+
},
|
| 1143 |
+
{
|
| 1144 |
+
"name": "bridge_dataset",
|
| 1145 |
+
"data_dir": null,
|
| 1146 |
+
"image_obs_keys": {
|
| 1147 |
+
"primary": "image_0",
|
| 1148 |
+
"high": null,
|
| 1149 |
+
"nav": null,
|
| 1150 |
+
"left_wrist": null,
|
| 1151 |
+
"right_wrist": null
|
| 1152 |
+
},
|
| 1153 |
+
"proprio_obs_keys": {
|
| 1154 |
+
"bimanual": null,
|
| 1155 |
+
"quadruped": null
|
| 1156 |
+
},
|
| 1157 |
+
"proprio_obs_dims": {
|
| 1158 |
+
"bimanual": 14,
|
| 1159 |
+
"quadruped": 59
|
| 1160 |
+
},
|
| 1161 |
+
"action_normalization_mask": [
|
| 1162 |
+
true,
|
| 1163 |
+
true,
|
| 1164 |
+
true,
|
| 1165 |
+
true,
|
| 1166 |
+
true,
|
| 1167 |
+
true,
|
| 1168 |
+
false
|
| 1169 |
+
],
|
| 1170 |
+
"language_key": "language_instruction",
|
| 1171 |
+
"action_proprio_normalization_type": "normal",
|
| 1172 |
+
"standardize_fn": {
|
| 1173 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 1174 |
+
"name": "bridge_dataset_transform",
|
| 1175 |
+
"args": [],
|
| 1176 |
+
"kwargs": {}
|
| 1177 |
+
}
|
| 1178 |
+
},
|
| 1179 |
+
{
|
| 1180 |
+
"name": "go1",
|
| 1181 |
+
"data_dir": null,
|
| 1182 |
+
"image_obs_keys": {
|
| 1183 |
+
"primary": null,
|
| 1184 |
+
"high": null,
|
| 1185 |
+
"nav": null,
|
| 1186 |
+
"left_wrist": null,
|
| 1187 |
+
"right_wrist": null
|
| 1188 |
+
},
|
| 1189 |
+
"proprio_obs_keys": {
|
| 1190 |
+
"bimanual": null,
|
| 1191 |
+
"quadruped": "proprio"
|
| 1192 |
+
},
|
| 1193 |
+
"proprio_obs_dims": {
|
| 1194 |
+
"bimanual": 14,
|
| 1195 |
+
"quadruped": 59
|
| 1196 |
+
},
|
| 1197 |
+
"action_normalization_mask": [
|
| 1198 |
+
true,
|
| 1199 |
+
true,
|
| 1200 |
+
true,
|
| 1201 |
+
true,
|
| 1202 |
+
true,
|
| 1203 |
+
true,
|
| 1204 |
+
true,
|
| 1205 |
+
true,
|
| 1206 |
+
true,
|
| 1207 |
+
true,
|
| 1208 |
+
true,
|
| 1209 |
+
true
|
| 1210 |
+
],
|
| 1211 |
+
"language_key": "language_instruction",
|
| 1212 |
+
"action_proprio_normalization_type": "normal",
|
| 1213 |
+
"standardize_fn": {
|
| 1214 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 1215 |
+
"name": "go1_dataset_transform",
|
| 1216 |
+
"args": [],
|
| 1217 |
+
"kwargs": {}
|
| 1218 |
+
}
|
| 1219 |
+
},
|
| 1220 |
+
{
|
| 1221 |
+
"name": "droid_wipe",
|
| 1222 |
+
"data_dir": null,
|
| 1223 |
+
"image_obs_keys": {
|
| 1224 |
+
"primary": "exterior_image_2_left",
|
| 1225 |
+
"high": null,
|
| 1226 |
+
"nav": null,
|
| 1227 |
+
"left_wrist": null,
|
| 1228 |
+
"right_wrist": null
|
| 1229 |
+
},
|
| 1230 |
+
"proprio_obs_keys": {
|
| 1231 |
+
"bimanual": null,
|
| 1232 |
+
"quadruped": null
|
| 1233 |
+
},
|
| 1234 |
+
"proprio_obs_dims": {
|
| 1235 |
+
"bimanual": 14,
|
| 1236 |
+
"quadruped": 59
|
| 1237 |
+
},
|
| 1238 |
+
"action_normalization_mask": [
|
| 1239 |
+
true,
|
| 1240 |
+
true,
|
| 1241 |
+
true,
|
| 1242 |
+
true,
|
| 1243 |
+
true,
|
| 1244 |
+
true,
|
| 1245 |
+
false
|
| 1246 |
+
],
|
| 1247 |
+
"language_key": "language_instruction",
|
| 1248 |
+
"action_proprio_normalization_type": "normal",
|
| 1249 |
+
"standardize_fn": {
|
| 1250 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 1251 |
+
"name": "droid_dataset_transform",
|
| 1252 |
+
"args": [],
|
| 1253 |
+
"kwargs": {}
|
| 1254 |
+
}
|
| 1255 |
+
},
|
| 1256 |
+
{
|
| 1257 |
+
"name": "omnimimic_gnm_dataset",
|
| 1258 |
+
"data_dir": null,
|
| 1259 |
+
"image_obs_keys": {
|
| 1260 |
+
"primary": null,
|
| 1261 |
+
"high": null,
|
| 1262 |
+
"nav": "image",
|
| 1263 |
+
"left_wrist": null,
|
| 1264 |
+
"right_wrist": null
|
| 1265 |
+
},
|
| 1266 |
+
"proprio_obs_keys": {
|
| 1267 |
+
"bimanual": null,
|
| 1268 |
+
"quadruped": null
|
| 1269 |
+
},
|
| 1270 |
+
"proprio_obs_dims": {
|
| 1271 |
+
"bimanual": 14,
|
| 1272 |
+
"quadruped": 59
|
| 1273 |
+
},
|
| 1274 |
+
"action_normalization_mask": [
|
| 1275 |
+
true,
|
| 1276 |
+
true
|
| 1277 |
+
],
|
| 1278 |
+
"language_key": "language_instruction",
|
| 1279 |
+
"action_proprio_normalization_type": "normal",
|
| 1280 |
+
"standardize_fn": {
|
| 1281 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 1282 |
+
"name": "omnimimic_gnm_transform",
|
| 1283 |
+
"args": [],
|
| 1284 |
+
"kwargs": {}
|
| 1285 |
+
}
|
| 1286 |
+
},
|
| 1287 |
+
{
|
| 1288 |
+
"name": "fractal20220817_data",
|
| 1289 |
+
"data_dir": null,
|
| 1290 |
+
"image_obs_keys": {
|
| 1291 |
+
"primary": "image",
|
| 1292 |
+
"high": null,
|
| 1293 |
+
"nav": null,
|
| 1294 |
+
"left_wrist": null,
|
| 1295 |
+
"right_wrist": null
|
| 1296 |
+
},
|
| 1297 |
+
"proprio_obs_keys": {
|
| 1298 |
+
"bimanual": null,
|
| 1299 |
+
"quadruped": null
|
| 1300 |
+
},
|
| 1301 |
+
"proprio_obs_dims": {
|
| 1302 |
+
"bimanual": 14,
|
| 1303 |
+
"quadruped": 59
|
| 1304 |
+
},
|
| 1305 |
+
"action_normalization_mask": [
|
| 1306 |
+
true,
|
| 1307 |
+
true,
|
| 1308 |
+
true,
|
| 1309 |
+
true,
|
| 1310 |
+
true,
|
| 1311 |
+
true,
|
| 1312 |
+
false
|
| 1313 |
+
],
|
| 1314 |
+
"language_key": "language_instruction",
|
| 1315 |
+
"action_proprio_normalization_type": "normal",
|
| 1316 |
+
"standardize_fn": {
|
| 1317 |
+
"module": "crossformer.data.oxe.oxe_standardization_transforms",
|
| 1318 |
+
"name": "rt1_dataset_transform",
|
| 1319 |
+
"args": [],
|
| 1320 |
+
"kwargs": {}
|
| 1321 |
+
}
|
| 1322 |
+
}
|
| 1323 |
+
],
|
| 1324 |
+
"frame_transform_kwargs": {
|
| 1325 |
+
"image_augment_kwargs": {
|
| 1326 |
+
"high": {
|
| 1327 |
+
"augment_order": [
|
| 1328 |
+
"random_resized_crop",
|
| 1329 |
+
"random_brightness",
|
| 1330 |
+
"random_contrast",
|
| 1331 |
+
"random_saturation",
|
| 1332 |
+
"random_hue"
|
| 1333 |
+
],
|
| 1334 |
+
"random_brightness": [
|
| 1335 |
+
0.1
|
| 1336 |
+
],
|
| 1337 |
+
"random_contrast": [
|
| 1338 |
+
0.9,
|
| 1339 |
+
1.1
|
| 1340 |
+
],
|
| 1341 |
+
"random_hue": [
|
| 1342 |
+
0.05
|
| 1343 |
+
],
|
| 1344 |
+
"random_resized_crop": {
|
| 1345 |
+
"ratio": [
|
| 1346 |
+
0.75,
|
| 1347 |
+
1.3333333333333333
|
| 1348 |
+
],
|
| 1349 |
+
"scale": [
|
| 1350 |
+
0.9,
|
| 1351 |
+
1.0
|
| 1352 |
+
]
|
| 1353 |
+
},
|
| 1354 |
+
"random_saturation": [
|
| 1355 |
+
0.9,
|
| 1356 |
+
1.1
|
| 1357 |
+
]
|
| 1358 |
+
},
|
| 1359 |
+
"left_wrist": {
|
| 1360 |
+
"augment_order": [
|
| 1361 |
+
"random_resized_crop",
|
| 1362 |
+
"random_brightness",
|
| 1363 |
+
"random_contrast",
|
| 1364 |
+
"random_saturation",
|
| 1365 |
+
"random_hue"
|
| 1366 |
+
],
|
| 1367 |
+
"random_brightness": [
|
| 1368 |
+
0.1
|
| 1369 |
+
],
|
| 1370 |
+
"random_contrast": [
|
| 1371 |
+
0.9,
|
| 1372 |
+
1.1
|
| 1373 |
+
],
|
| 1374 |
+
"random_hue": [
|
| 1375 |
+
0.05
|
| 1376 |
+
],
|
| 1377 |
+
"random_resized_crop": {
|
| 1378 |
+
"ratio": [
|
| 1379 |
+
0.75,
|
| 1380 |
+
1.3333333333333333
|
| 1381 |
+
],
|
| 1382 |
+
"scale": [
|
| 1383 |
+
0.9,
|
| 1384 |
+
1.0
|
| 1385 |
+
]
|
| 1386 |
+
},
|
| 1387 |
+
"random_saturation": [
|
| 1388 |
+
0.9,
|
| 1389 |
+
1.1
|
| 1390 |
+
]
|
| 1391 |
+
},
|
| 1392 |
+
"nav": {
|
| 1393 |
+
"augment_order": [
|
| 1394 |
+
"random_resized_crop",
|
| 1395 |
+
"random_brightness",
|
| 1396 |
+
"random_contrast",
|
| 1397 |
+
"random_saturation",
|
| 1398 |
+
"random_hue"
|
| 1399 |
+
],
|
| 1400 |
+
"random_brightness": [
|
| 1401 |
+
0.1
|
| 1402 |
+
],
|
| 1403 |
+
"random_contrast": [
|
| 1404 |
+
0.9,
|
| 1405 |
+
1.1
|
| 1406 |
+
],
|
| 1407 |
+
"random_hue": [
|
| 1408 |
+
0.05
|
| 1409 |
+
],
|
| 1410 |
+
"random_resized_crop": {
|
| 1411 |
+
"ratio": [
|
| 1412 |
+
0.9,
|
| 1413 |
+
1.1
|
| 1414 |
+
],
|
| 1415 |
+
"scale": [
|
| 1416 |
+
0.8,
|
| 1417 |
+
1.0
|
| 1418 |
+
]
|
| 1419 |
+
},
|
| 1420 |
+
"random_saturation": [
|
| 1421 |
+
0.9,
|
| 1422 |
+
1.1
|
| 1423 |
+
]
|
| 1424 |
+
},
|
| 1425 |
+
"primary": {
|
| 1426 |
+
"augment_order": [
|
| 1427 |
+
"random_resized_crop",
|
| 1428 |
+
"random_brightness",
|
| 1429 |
+
"random_contrast",
|
| 1430 |
+
"random_saturation",
|
| 1431 |
+
"random_hue"
|
| 1432 |
+
],
|
| 1433 |
+
"random_brightness": [
|
| 1434 |
+
0.1
|
| 1435 |
+
],
|
| 1436 |
+
"random_contrast": [
|
| 1437 |
+
0.9,
|
| 1438 |
+
1.1
|
| 1439 |
+
],
|
| 1440 |
+
"random_hue": [
|
| 1441 |
+
0.05
|
| 1442 |
+
],
|
| 1443 |
+
"random_resized_crop": {
|
| 1444 |
+
"ratio": [
|
| 1445 |
+
0.9,
|
| 1446 |
+
1.1
|
| 1447 |
+
],
|
| 1448 |
+
"scale": [
|
| 1449 |
+
0.8,
|
| 1450 |
+
1.0
|
| 1451 |
+
]
|
| 1452 |
+
},
|
| 1453 |
+
"random_saturation": [
|
| 1454 |
+
0.9,
|
| 1455 |
+
1.1
|
| 1456 |
+
]
|
| 1457 |
+
},
|
| 1458 |
+
"right_wrist": {
|
| 1459 |
+
"augment_order": [
|
| 1460 |
+
"random_resized_crop",
|
| 1461 |
+
"random_brightness",
|
| 1462 |
+
"random_contrast",
|
| 1463 |
+
"random_saturation",
|
| 1464 |
+
"random_hue"
|
| 1465 |
+
],
|
| 1466 |
+
"random_brightness": [
|
| 1467 |
+
0.1
|
| 1468 |
+
],
|
| 1469 |
+
"random_contrast": [
|
| 1470 |
+
0.9,
|
| 1471 |
+
1.1
|
| 1472 |
+
],
|
| 1473 |
+
"random_hue": [
|
| 1474 |
+
0.05
|
| 1475 |
+
],
|
| 1476 |
+
"random_resized_crop": {
|
| 1477 |
+
"ratio": [
|
| 1478 |
+
0.75,
|
| 1479 |
+
1.3333333333333333
|
| 1480 |
+
],
|
| 1481 |
+
"scale": [
|
| 1482 |
+
0.9,
|
| 1483 |
+
1.0
|
| 1484 |
+
]
|
| 1485 |
+
},
|
| 1486 |
+
"random_saturation": [
|
| 1487 |
+
0.9,
|
| 1488 |
+
1.1
|
| 1489 |
+
]
|
| 1490 |
+
}
|
| 1491 |
+
},
|
| 1492 |
+
"num_parallel_calls": 200,
|
| 1493 |
+
"resize_size": {
|
| 1494 |
+
"high": [
|
| 1495 |
+
224,
|
| 1496 |
+
224
|
| 1497 |
+
],
|
| 1498 |
+
"left_wrist": [
|
| 1499 |
+
224,
|
| 1500 |
+
224
|
| 1501 |
+
],
|
| 1502 |
+
"nav": [
|
| 1503 |
+
224,
|
| 1504 |
+
224
|
| 1505 |
+
],
|
| 1506 |
+
"primary": [
|
| 1507 |
+
224,
|
| 1508 |
+
224
|
| 1509 |
+
],
|
| 1510 |
+
"right_wrist": [
|
| 1511 |
+
224,
|
| 1512 |
+
224
|
| 1513 |
+
]
|
| 1514 |
+
}
|
| 1515 |
+
},
|
| 1516 |
+
"sample_weights": [
|
| 1517 |
+
0.021755552811740044,
|
| 1518 |
+
0.009986998241552502,
|
| 1519 |
+
0.0023179438302583103,
|
| 1520 |
+
0.0016371801232614254,
|
| 1521 |
+
0.0008888419195460733,
|
| 1522 |
+
0.007874885740329124,
|
| 1523 |
+
0.0032054418194781,
|
| 1524 |
+
0.004113224757060238,
|
| 1525 |
+
0.006839324427193265,
|
| 1526 |
+
0.01479473259297815,
|
| 1527 |
+
0.015045048015813084,
|
| 1528 |
+
0.0007163148639093663,
|
| 1529 |
+
0.002826978440249719,
|
| 1530 |
+
0.008290489894059044,
|
| 1531 |
+
0.007414589603576858,
|
| 1532 |
+
0.005878414243900214,
|
| 1533 |
+
0.021755552811740044,
|
| 1534 |
+
0.0001874782805166165,
|
| 1535 |
+
0.0030709017944702635,
|
| 1536 |
+
0.007599149035416076,
|
| 1537 |
+
0.0026296096724881066,
|
| 1538 |
+
0.0005253087662862543,
|
| 1539 |
+
0.0002175555281174004,
|
| 1540 |
+
0.085,
|
| 1541 |
+
0.014166666666666666,
|
| 1542 |
+
0.014166666666666666,
|
| 1543 |
+
0.014166666666666666,
|
| 1544 |
+
0.014166666666666666,
|
| 1545 |
+
0.014166666666666666,
|
| 1546 |
+
0.014166666666666666,
|
| 1547 |
+
0.17,
|
| 1548 |
+
0.085,
|
| 1549 |
+
0.085,
|
| 1550 |
+
0.17,
|
| 1551 |
+
0.17
|
| 1552 |
+
],
|
| 1553 |
+
"shuffle_buffer_size": 25000,
|
| 1554 |
+
"traj_read_threads": 48,
|
| 1555 |
+
"traj_transform_kwargs": {
|
| 1556 |
+
"action_horizon": 100,
|
| 1557 |
+
"goal_relabeling_kwargs": {
|
| 1558 |
+
"max_goal_distance": 15
|
| 1559 |
+
},
|
| 1560 |
+
"goal_relabeling_strategy": "uniform",
|
| 1561 |
+
"head_to_dataset": {
|
| 1562 |
+
"bimanual": [
|
| 1563 |
+
"aloha_pen_uncap_diverse_dataset",
|
| 1564 |
+
"aloha_dough_cut_dataset",
|
| 1565 |
+
"aloha_lucy_dataset",
|
| 1566 |
+
"aloha_drawer_dataset",
|
| 1567 |
+
"aloha_pick_place_dataset",
|
| 1568 |
+
"aloha_static_dataset",
|
| 1569 |
+
"aloha_sushi_cut_full_dataset"
|
| 1570 |
+
],
|
| 1571 |
+
"nav": [
|
| 1572 |
+
"omnimimic_gnm_dataset"
|
| 1573 |
+
],
|
| 1574 |
+
"quadruped": [
|
| 1575 |
+
"go1"
|
| 1576 |
+
],
|
| 1577 |
+
"single_arm": [
|
| 1578 |
+
"bridge_dataset",
|
| 1579 |
+
"fractal20220817_data",
|
| 1580 |
+
"kuka",
|
| 1581 |
+
"taco_play",
|
| 1582 |
+
"taco_extra",
|
| 1583 |
+
"jaco_play",
|
| 1584 |
+
"berkeley_cable_routing",
|
| 1585 |
+
"roboturk",
|
| 1586 |
+
"nyu_door_opening_surprising_effectiveness",
|
| 1587 |
+
"viola",
|
| 1588 |
+
"berkeley_autolab_ur5",
|
| 1589 |
+
"toto",
|
| 1590 |
+
"language_table",
|
| 1591 |
+
"stanford_hydra_dataset_converted_externally_to_rlds",
|
| 1592 |
+
"austin_buds_dataset_converted_externally_to_rlds",
|
| 1593 |
+
"nyu_franka_play_dataset_converted_externally_to_rlds",
|
| 1594 |
+
"furniture_bench_dataset_converted_externally_to_rlds",
|
| 1595 |
+
"austin_sailor_dataset_converted_externally_to_rlds",
|
| 1596 |
+
"austin_sirius_dataset_converted_externally_to_rlds",
|
| 1597 |
+
"bc_z",
|
| 1598 |
+
"dlr_edan_shared_control_converted_externally_to_rlds",
|
| 1599 |
+
"iamlab_cmu_pickup_insert_converted_externally_to_rlds",
|
| 1600 |
+
"utaustin_mutex",
|
| 1601 |
+
"berkeley_fanuc_manipulation",
|
| 1602 |
+
"cmu_stretch",
|
| 1603 |
+
"droid",
|
| 1604 |
+
"droid_wipe"
|
| 1605 |
+
]
|
| 1606 |
+
},
|
| 1607 |
+
"max_action_dim": 14,
|
| 1608 |
+
"subsample_length": 100,
|
| 1609 |
+
"task_augment_kwargs": {
|
| 1610 |
+
"keep_image_prob": 1.0
|
| 1611 |
+
},
|
| 1612 |
+
"task_augment_strategy": "delete_task_conditioning",
|
| 1613 |
+
"window_size": 5
|
| 1614 |
+
},
|
| 1615 |
+
"traj_transform_threads": 48
|
| 1616 |
+
},
|
| 1617 |
+
"eval_datasets": [],
|
| 1618 |
+
"eval_interval": 5e+20,
|
| 1619 |
+
"log_interval": 500,
|
| 1620 |
+
"model": {
|
| 1621 |
+
"heads": {
|
| 1622 |
+
"bimanual": {
|
| 1623 |
+
"args": [],
|
| 1624 |
+
"kwargs": {
|
| 1625 |
+
"action_dim": 14,
|
| 1626 |
+
"action_horizon": 100,
|
| 1627 |
+
"clip_pred": false,
|
| 1628 |
+
"constrain_loss_dims": true,
|
| 1629 |
+
"loss_weight": 1.0,
|
| 1630 |
+
"num_preds": 14,
|
| 1631 |
+
"pool_strategy": "pass",
|
| 1632 |
+
"readout_key": "readout_bimanual"
|
| 1633 |
+
},
|
| 1634 |
+
"module": "crossformer.model.components.action_heads",
|
| 1635 |
+
"name": "L1ActionHead"
|
| 1636 |
+
},
|
| 1637 |
+
"nav": {
|
| 1638 |
+
"args": [],
|
| 1639 |
+
"kwargs": {
|
| 1640 |
+
"action_dim": 2,
|
| 1641 |
+
"action_horizon": 4,
|
| 1642 |
+
"clip_pred": false,
|
| 1643 |
+
"constrain_loss_dims": true,
|
| 1644 |
+
"loss_weight": 1.0,
|
| 1645 |
+
"num_preds": 2,
|
| 1646 |
+
"pool_strategy": "pass",
|
| 1647 |
+
"readout_key": "readout_nav"
|
| 1648 |
+
},
|
| 1649 |
+
"module": "crossformer.model.components.action_heads",
|
| 1650 |
+
"name": "L1ActionHead"
|
| 1651 |
+
},
|
| 1652 |
+
"quadruped": {
|
| 1653 |
+
"args": [],
|
| 1654 |
+
"kwargs": {
|
| 1655 |
+
"action_dim": 12,
|
| 1656 |
+
"action_horizon": 1,
|
| 1657 |
+
"clip_pred": false,
|
| 1658 |
+
"constrain_loss_dims": true,
|
| 1659 |
+
"loss_weight": 1.0,
|
| 1660 |
+
"num_preds": 12,
|
| 1661 |
+
"pool_strategy": "pass",
|
| 1662 |
+
"readout_key": "readout_quadruped"
|
| 1663 |
+
},
|
| 1664 |
+
"module": "crossformer.model.components.action_heads",
|
| 1665 |
+
"name": "L1ActionHead"
|
| 1666 |
+
},
|
| 1667 |
+
"single_arm": {
|
| 1668 |
+
"args": [],
|
| 1669 |
+
"kwargs": {
|
| 1670 |
+
"action_dim": 7,
|
| 1671 |
+
"action_horizon": 4,
|
| 1672 |
+
"clip_pred": false,
|
| 1673 |
+
"constrain_loss_dims": true,
|
| 1674 |
+
"loss_weight": 1.0,
|
| 1675 |
+
"num_preds": 7,
|
| 1676 |
+
"pool_strategy": "pass",
|
| 1677 |
+
"readout_key": "readout_single_arm"
|
| 1678 |
+
},
|
| 1679 |
+
"module": "crossformer.model.components.action_heads",
|
| 1680 |
+
"name": "L1ActionHead"
|
| 1681 |
+
}
|
| 1682 |
+
},
|
| 1683 |
+
"max_horizon": 10,
|
| 1684 |
+
"observation_tokenizers": {
|
| 1685 |
+
"bimanual": {
|
| 1686 |
+
"args": [],
|
| 1687 |
+
"kwargs": {
|
| 1688 |
+
"dropout_rate": 0.2,
|
| 1689 |
+
"obs_keys": [
|
| 1690 |
+
"proprio_bimanual"
|
| 1691 |
+
]
|
| 1692 |
+
},
|
| 1693 |
+
"module": "crossformer.model.components.tokenizers",
|
| 1694 |
+
"name": "LowdimObsTokenizer"
|
| 1695 |
+
},
|
| 1696 |
+
"high": {
|
| 1697 |
+
"args": [],
|
| 1698 |
+
"kwargs": {
|
| 1699 |
+
"encoder": {
|
| 1700 |
+
"args": [],
|
| 1701 |
+
"kwargs": {},
|
| 1702 |
+
"module": "crossformer.model.components.vit_encoders",
|
| 1703 |
+
"name": "ResNet26FILM"
|
| 1704 |
+
},
|
| 1705 |
+
"obs_stack_keys": [
|
| 1706 |
+
"image_high"
|
| 1707 |
+
],
|
| 1708 |
+
"task_film_keys": [
|
| 1709 |
+
"language_instruction"
|
| 1710 |
+
],
|
| 1711 |
+
"task_stack_keys": [
|
| 1712 |
+
"image_high"
|
| 1713 |
+
]
|
| 1714 |
+
},
|
| 1715 |
+
"module": "crossformer.model.components.tokenizers",
|
| 1716 |
+
"name": "ImageTokenizer"
|
| 1717 |
+
},
|
| 1718 |
+
"left": {
|
| 1719 |
+
"args": [],
|
| 1720 |
+
"kwargs": {
|
| 1721 |
+
"encoder": {
|
| 1722 |
+
"args": [],
|
| 1723 |
+
"kwargs": {},
|
| 1724 |
+
"module": "crossformer.model.components.vit_encoders",
|
| 1725 |
+
"name": "ResNet26FILM"
|
| 1726 |
+
},
|
| 1727 |
+
"obs_stack_keys": [
|
| 1728 |
+
"image_left_wrist"
|
| 1729 |
+
],
|
| 1730 |
+
"task_film_keys": [
|
| 1731 |
+
"language_instruction"
|
| 1732 |
+
],
|
| 1733 |
+
"task_stack_keys": []
|
| 1734 |
+
},
|
| 1735 |
+
"module": "crossformer.model.components.tokenizers",
|
| 1736 |
+
"name": "ImageTokenizer"
|
| 1737 |
+
},
|
| 1738 |
+
"nav": {
|
| 1739 |
+
"args": [],
|
| 1740 |
+
"kwargs": {
|
| 1741 |
+
"encoder": {
|
| 1742 |
+
"args": [],
|
| 1743 |
+
"kwargs": {},
|
| 1744 |
+
"module": "crossformer.model.components.vit_encoders",
|
| 1745 |
+
"name": "ResNet26"
|
| 1746 |
+
},
|
| 1747 |
+
"obs_stack_keys": [
|
| 1748 |
+
"image_nav"
|
| 1749 |
+
],
|
| 1750 |
+
"task_film_keys": [],
|
| 1751 |
+
"task_stack_keys": [
|
| 1752 |
+
"image_nav"
|
| 1753 |
+
]
|
| 1754 |
+
},
|
| 1755 |
+
"module": "crossformer.model.components.tokenizers",
|
| 1756 |
+
"name": "ImageTokenizer"
|
| 1757 |
+
},
|
| 1758 |
+
"primary": {
|
| 1759 |
+
"args": [],
|
| 1760 |
+
"kwargs": {
|
| 1761 |
+
"encoder": {
|
| 1762 |
+
"args": [],
|
| 1763 |
+
"kwargs": {},
|
| 1764 |
+
"module": "crossformer.model.components.vit_encoders",
|
| 1765 |
+
"name": "ResNet26FILM"
|
| 1766 |
+
},
|
| 1767 |
+
"obs_stack_keys": [
|
| 1768 |
+
"image_primary"
|
| 1769 |
+
],
|
| 1770 |
+
"task_film_keys": [
|
| 1771 |
+
"language_instruction"
|
| 1772 |
+
],
|
| 1773 |
+
"task_stack_keys": [
|
| 1774 |
+
"image_primary"
|
| 1775 |
+
]
|
| 1776 |
+
},
|
| 1777 |
+
"module": "crossformer.model.components.tokenizers",
|
| 1778 |
+
"name": "ImageTokenizer"
|
| 1779 |
+
},
|
| 1780 |
+
"quadruped": {
|
| 1781 |
+
"args": [],
|
| 1782 |
+
"kwargs": {
|
| 1783 |
+
"obs_keys": [
|
| 1784 |
+
"proprio_quadruped"
|
| 1785 |
+
]
|
| 1786 |
+
},
|
| 1787 |
+
"module": "crossformer.model.components.tokenizers",
|
| 1788 |
+
"name": "LowdimObsTokenizer"
|
| 1789 |
+
},
|
| 1790 |
+
"right": {
|
| 1791 |
+
"args": [],
|
| 1792 |
+
"kwargs": {
|
| 1793 |
+
"encoder": {
|
| 1794 |
+
"args": [],
|
| 1795 |
+
"kwargs": {},
|
| 1796 |
+
"module": "crossformer.model.components.vit_encoders",
|
| 1797 |
+
"name": "ResNet26FILM"
|
| 1798 |
+
},
|
| 1799 |
+
"obs_stack_keys": [
|
| 1800 |
+
"image_right_wrist"
|
| 1801 |
+
],
|
| 1802 |
+
"task_film_keys": [
|
| 1803 |
+
"language_instruction"
|
| 1804 |
+
],
|
| 1805 |
+
"task_stack_keys": []
|
| 1806 |
+
},
|
| 1807 |
+
"module": "crossformer.model.components.tokenizers",
|
| 1808 |
+
"name": "ImageTokenizer"
|
| 1809 |
+
}
|
| 1810 |
+
},
|
| 1811 |
+
"readouts": {
|
| 1812 |
+
"bimanual": 100,
|
| 1813 |
+
"nav": 4,
|
| 1814 |
+
"quadruped": 1,
|
| 1815 |
+
"single_arm": 4
|
| 1816 |
+
},
|
| 1817 |
+
"repeat_task_tokens": true,
|
| 1818 |
+
"task_tokenizers": {},
|
| 1819 |
+
"token_embedding_size": 512,
|
| 1820 |
+
"transformer_kwargs": {
|
| 1821 |
+
"attention_dropout_rate": 0.0,
|
| 1822 |
+
"dropout_rate": 0.1,
|
| 1823 |
+
"mlp_dim": 2048,
|
| 1824 |
+
"num_attention_heads": 8,
|
| 1825 |
+
"num_layers": 12,
|
| 1826 |
+
"repeat_pos_enc": true
|
| 1827 |
+
},
|
| 1828 |
+
"use_correct_attention": true
|
| 1829 |
+
},
|
| 1830 |
+
"num_steps": 300000,
|
| 1831 |
+
"optimizer": {
|
| 1832 |
+
"clip_gradient": 1.0,
|
| 1833 |
+
"frozen_keys": [],
|
| 1834 |
+
"learning_rate": {
|
| 1835 |
+
"init_value": 0.0,
|
| 1836 |
+
"name": "rsqrt",
|
| 1837 |
+
"peak_value": 0.0003,
|
| 1838 |
+
"timescale": 10000,
|
| 1839 |
+
"warmup_steps": 2000
|
| 1840 |
+
},
|
| 1841 |
+
"weight_decay": 0.1
|
| 1842 |
+
},
|
| 1843 |
+
"prefetch_num_batches": 0,
|
| 1844 |
+
"pretrained_loaders": [
|
| 1845 |
+
{
|
| 1846 |
+
"module": "experiments.sudeep.aloha.resnet_pt",
|
| 1847 |
+
"name": "resnet_26_loader",
|
| 1848 |
+
"args": [],
|
| 1849 |
+
"kwargs": {
|
| 1850 |
+
"restore_path": ""
|
| 1851 |
+
}
|
| 1852 |
+
}
|
| 1853 |
+
],
|
| 1854 |
+
"resume_path": null,
|
| 1855 |
+
"save_dir": null,
|
| 1856 |
+
"save_interval": 10000,
|
| 1857 |
+
"seed": 42,
|
| 1858 |
+
"skip_norm_keys": [
|
| 1859 |
+
"proprio_bimanual"
|
| 1860 |
+
],
|
| 1861 |
+
"start_step": null,
|
| 1862 |
+
"text_processor": {
|
| 1863 |
+
"args": [],
|
| 1864 |
+
"kwargs": {},
|
| 1865 |
+
"module": "crossformer.data.utils.text_processing",
|
| 1866 |
+
"name": "UniversalSentenceEncoder"
|
| 1867 |
+
},
|
| 1868 |
+
"val_kwargs": {
|
| 1869 |
+
"num_val_batches": 16,
|
| 1870 |
+
"val_shuffle_buffer_size": 1000
|
| 1871 |
+
},
|
| 1872 |
+
"viz_datasets": [],
|
| 1873 |
+
"viz_interval": 5e+20,
|
| 1874 |
+
"viz_kwargs": {
|
| 1875 |
+
"eval_batch_size": 128,
|
| 1876 |
+
"samples_per_state": 8,
|
| 1877 |
+
"trajs_for_metrics": 100,
|
| 1878 |
+
"trajs_for_viz": 8
|
| 1879 |
+
},
|
| 1880 |
+
"wandb": {
|
| 1881 |
+
"entity": null,
|
| 1882 |
+
"group": null,
|
| 1883 |
+
"project": "crossformer"
|
| 1884 |
+
},
|
| 1885 |
+
"wandb_resume_id": null,
|
| 1886 |
+
"window_size": 5
|
| 1887 |
+
}
|
dataset_statistics.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"aloha_dough_cut_dataset": {"action": {"mask": [true, true, true, true, true, true, false, true, true, true, true, true, true, false], "max": [0.9035146832466125, 0.5445631742477417, 1.4726215600967407, 0.7117670774459839, 1.1228740215301514, 1.6260197162628174, 1.038834810256958, 0.79460209608078, 0.6412039995193481, 1.372912883758545, 1.5079030990600586, 1.5769323110580444, 1.6321556568145752, 1.0695143938064575], "mean": [-0.06304754316806793, -0.0798826515674591, 0.9165536761283875, -0.5000047087669373, -0.5068535208702087, 0.1275118738412857, 0.6355013847351074, 0.025164712220430374, 0.06502483040094376, 0.5596698522567749, -0.030039390549063683, 0.03929644078016281, 0.11276345700025558, 0.5265209078788757], "min": [-0.6396700143814087, -1.0108933448791504, -0.12271846830844879, -1.8223692178726196, -1.4281361103057861, -1.0661166906356812, -0.09631098061800003, -0.424912691116333, -0.9587380290031433, -0.905048668384552, -1.3897866010665894, -1.2640001773834229, -0.598252534866333, -0.099920354783535], "p01": [-0.4893398880958557, -0.8697671294212341, 0.19788353145122528, -1.4848934412002563, -1.0737866163253784, -0.8390874862670898, 0.0029466012492775917, -0.374291330575943, -0.6366020441055298, -0.2761165499687195, -0.7531846165657043, -0.9786797761917114, -0.3880971372127533, -0.05660797655582428], "p99": [0.6826214790344238, 0.44792240858078003, 1.402058482170105, 0.3574175238609314, 0.7516506314277649, 0.826815664768219, 0.9846943020820618, 0.4494563937187195, 0.4862719178199768, 1.1428157091140747, 0.8682331442832947, 0.9756118059158325, 0.9142525792121887, 1.008155107498169], "std": [0.3099515438079834, 0.3811430037021637, 0.2682836949825287, 0.394284188747406, 0.3991157114505768, 0.361807644367218, 0.20688240230083466, 0.1902119517326355, 0.2621226906776428, 0.3672671914100647, 0.3097095787525177, 0.4848240613937378, 0.24292759597301483, 0.3102242052555084]}, "num_trajectories": 750, "num_transitions": 323368, "proprio_bimanual": {"max": [0.905048668384552, 0.5123496055603027, 1.483359456062317, 0.7025632262229919, 1.1228740215301514, 1.6229517459869385, 0.9999999403953552, 0.7915341258049011, 0.676485538482666, 1.382116675376892, 1.483359456062317, 1.575398325920105, 1.6321556568145752, 0.9999999403953552], "mean": [-0.06225714832544327, -0.07485871016979218, 0.9303997755050659, -0.5010020732879639, -0.506401777267456, 0.12651260197162628, 0.6978353261947632, 0.02803533710539341, 0.08266624808311462, 0.5723478198051453, -0.030892668291926384, 0.04010569676756859, 0.11194124817848206, 0.640129029750824], "min": [-0.6366020441055298, -1.0139613151550293, -0.08590292930603027, -1.8208352327346802, -1.4066604375839233, -1.0661166906356812, 0.11511506885290146, -0.42031073570251465, -0.9710098505020142, -0.8759030699729919, -1.3805826902389526, -1.2517284154891968, -0.5967185497283936, 0.28998684883117676], "p01": [-0.48780590295791626, -0.8881748914718628, 0.21935926377773285, -1.4876853227615356, -1.0661166906356812, -0.8406214714050293, 0.12229029834270477, -0.3712233603000641, -0.7148350477218628, -0.2592427730560303, -0.7501166462898254, -0.9694758653640747, -0.3880971372127533, 0.30347827076911926], "p99": [0.6810874938964844, 0.42644667625427246, 1.4173983335494995, 0.3543495833873749, 0.7516506314277649, 0.826815664768219, 0.994238018989563, 0.45099037885665894, 0.5169515609741211, 1.1658254861831665, 0.8682331442832947, 0.9802137613296509, 0.9127185940742493, 0.9981253743171692], "std": [0.3082458972930908, 0.3840135633945465, 0.2692725360393524, 0.3947542607784271, 0.3983606994152069, 0.36215123534202576, 0.22308988869190216, 0.19022829830646515, 0.2655238211154938, 0.36651065945625305, 0.3084445893764496, 0.48738494515419006, 0.24280770123004913, 0.24256202578544617]}}, "aloha_drawer_dataset": {"action": {"mask": [true, true, true, true, true, true, false, true, true, true, true, true, true, false], "max": [0.6320000886917114, 0.4279806613922119, 1.4465439319610596, 0.41724279522895813, 0.6580777764320374, 0.7455146908760071, 1.0659050941467285, 0.7792622447013855, 1.087592363357544, 1.38825261592865, 2.552544116973877, 1.9113401174545288, 3.103243112564087, 1.1182408332824707], "mean": [-0.027892552316188812, -0.25454142689704895, 0.9634109139442444, -0.11814866214990616, -0.3652156591415405, 0.04810073599219322, 0.45852425694465637, -0.05010424926877022, -0.17347979545593262, 0.6589820981025696, -0.06797533482313156, 0.43668216466903687, 0.3321996033191681, 0.5356459617614746], "min": [-0.5123496055603027, -1.0339031219482422, 0.374291330575943, -0.8421554565429688, -1.0477088689804077, -1.6919808387756348, -0.09811566770076752, -0.9710098505020142, -1.115204095840454, -2.2380781173706055, -3.130854845046997, -1.6352235078811646, -1.0615147352218628, -0.07645954936742783], "p01": [-0.3267379105091095, -0.9480001330375671, 0.5123496055603027, -0.7025632262229919, -0.9464661478996277, -0.015339808538556099, -0.03314702585339546, -0.7240389585494995, -0.8805050253868103, -0.2960582971572876, -1.7824857234954834, -0.37275734543800354, -0.5829127430915833, 0.02821209840476513], "p99": [0.40343695878982544, 0.2853204309940338, 1.3422331809997559, 0.18867963552474976, 0.2162145972251892, 0.5568350553512573, 1.0207879543304443, 0.4141748249530792, 0.3988350033760071, 1.1581555604934692, 0.7010292410850525, 1.5677284002304077, 2.287165403366089, 1.038834810256958], "std": [0.15365876257419586, 0.3653314411640167, 0.22074660658836365, 0.16818614304065704, 0.24236515164375305, 0.13198280334472656, 0.3341043293476105, 0.27787140011787415, 0.27358320355415344, 0.34550976753234863, 0.4653882384300232, 0.42368364334106445, 0.6024450063705444, 0.28179994225502014]}, "num_trajectories": 1081, "num_transitions": 476206, "proprio_bimanual": {"max": [0.6289321184158325, 0.3942330777645111, 1.4603497982025146, 0.4141748249530792, 0.647339940071106, 0.7439807057380676, 0.9999999403953552, 0.7761942744255066, 0.8958448171615601, 1.3974565267562866, 2.549476146697998, 1.9098061323165894, 3.0986413955688477, 0.9999999403953552], "mean": [-0.027400396764278412, -0.2504436671733856, 0.9803711175918579, -0.11841507256031036, -0.3596595227718353, 0.04704941064119339, 0.5113952159881592, -0.04758168384432793, -0.15455181896686554, 0.676523745059967, -0.06775331497192383, 0.4392300844192505, 0.33117926120758057, 0.6269031167030334], "min": [-0.5154175758361816, -1.0354371070861816, 0.4049709439277649, -0.8421554565429688, -1.0246992111206055, -1.6950488090515137, 0.11318255215883255, -0.9664078950881958, -1.121340036392212, -1.6935148239135742, -3.1216509342193604, -1.6306216716766357, -1.0599807500839233, 0.2837609052658081], "p01": [-0.31753402948379517, -0.954136073589325, 0.5322913527488708, -0.7040972113609314, -0.9357283115386963, -0.016873789951205254, 0.11705803871154785, -0.7225049734115601, -0.9081166386604309, -0.2761165499687195, -1.7840197086334229, -0.36178937554359436, -0.5813787579536438, 0.3107548654079437], "p99": [0.40343695878982544, 0.2822524607181549, 1.3606410026550293, 0.18561168015003204, 0.21322333812713623, 0.5568350553512573, 0.9993929266929626, 0.4141748249530792, 0.43104860186576843, 1.1765632629394531, 0.6979612708091736, 1.5677284002304077, 2.287165403366089, 0.9999483823776245], "std": [0.15066370368003845, 0.371857613325119, 0.22081804275512695, 0.16783887147903442, 0.24046136438846588, 0.13172966241836548, 0.33406326174736023, 0.2774706482887268, 0.28276026248931885, 0.3440132737159729, 0.46372511982917786, 0.42298591136932373, 0.6014875173568726, 0.258215069770813]}}, "aloha_lucy_dataset": {"action": {"mask": [true, true, true, true, true, true, false, true, true, true, true, true, true, false], "max": [1.0921943187713623, 0.6657477021217346, 1.558524489402771, 1.9757672548294067, 1.9496896266937256, 1.3637089729309082, 1.1182408332824707, 1.5109710693359375, 1.023165225982666, 1.5631264448165894, 1.320757508277893, 2.11075758934021, 3.5465636253356934, 1.1146315336227417], "mean": [0.08728872239589691, -0.26688215136528015, 0.6374562382698059, 0.10192633420228958, 0.5537572503089905, -0.3897952735424042, 0.11758702248334885, 0.14259926974773407, -0.26143139600753784, 0.7681571245193481, -0.5416339039802551, 0.5731332898139954, 1.0158259868621826, 0.16693587601184845], "min": [-0.5215534567832947, -1.3437671661376953, -0.7347767949104309, -1.5968739986419678, -1.398990511894226, -1.7518061399459839, -0.16669368743896484, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -1.7548149824142456], "p01": [-0.3190680146217346, -1.2333205938339233, -0.0475534051656723, -0.7271069288253784, -0.3344078063964844, -1.2409905195236206, -0.0060768406838178635, -0.7869321703910828, -1.0599807500839233, -0.25310683250427246, -1.6873788833618164, -0.3267379105091095, -0.003067961661145091, -0.011490905657410622], "p99": [0.6948933005332947, 0.27458256483078003, 1.3146215677261353, 1.136679768562317, 1.257864236831665, 0.2822524607181549, 0.9864989519119263, 1.2379225492477417, 0.6028544902801514, 1.3222914934158325, 0.4832039475440979, 1.477223515510559, 2.4666411876678467, 0.9991317987442017], "std": [0.23480872809886932, 0.39629220962524414, 0.3926001191139221, 0.325132817029953, 0.3607615828514099, 0.31728941202163696, 0.2358071357011795, 0.46482887864112854, 0.42249712347984314, 0.3832353949546814, 0.5242871046066284, 0.4726457893848419, 0.7154117822647095, 0.2799983322620392]}, "num_trajectories": 13499, "num_transitions": 2771613, "proprio_bimanual": {"max": [1.0860583782196045, 0.6887574195861816, 1.5646604299545288, 1.9665634632110596, 1.9420197010040283, 1.3621749877929688, 0.9999999403953552, 1.5109710693359375, 1.055378794670105, 1.5692623853683472, 1.3176895380020142, 2.100019693374634, 3.1431267261505127, 0.9999999403953552], "mean": [0.0875844731926918, -0.2556387484073639, 0.6545287370681763, 0.10036227107048035, 0.5572015643119812, -0.3893788754940033, 0.13716308772563934, 0.14305485785007477, -0.24726520478725433, 0.7851355671882629, -0.5382525324821472, 0.5736652612686157, 1.0144864320755005, 0.398139625787735], "min": [-0.52001953125, -1.3499031066894531, -0.7117670774459839, -1.5999419689178467, -1.38825261592865, -1.745670199394226, -0.02406238205730915, -1.3836506605148315, -1.1919031143188477, -1.136679768562317, -3.0004665851593018, -1.1520196199417114, -0.8283496499061584, -0.033567335456609726], "p01": [-0.31753402948379517, -1.2501944303512573, -0.029145635664463043, -0.7271069288253784, -0.3267379105091095, -1.2394565343856812, 0.008260209113359451, -0.7823302149772644, -1.0676506757736206, -0.2346990704536438, -1.6797089576721191, -0.3206019997596741, -0.003067961661145091, 0.0023537499364465475], "p99": [0.6948933005332947, 0.30066025257110596, 1.3268934488296509, 1.1290098428726196, 1.2593982219696045, 0.2791845202445984, 0.9943925142288208, 1.2363885641098022, 0.6381360292434692, 1.3345633745193481, 0.48166999220848083, 1.4741555452346802, 2.4651072025299072, 0.9969291090965271], "std": [0.23456239700317383, 0.4076337516307831, 0.39144784212112427, 0.323933869600296, 0.36007335782051086, 0.3169434666633606, 0.24385134875774384, 0.46299219131469727, 0.43540143966674805, 0.3822689354419708, 0.5215927362442017, 0.4705725610256195, 0.714769184589386, 0.27767062187194824]}}, "aloha_pen_uncap_diverse_dataset": {"action": {"mask": [true, true, true, true, true, true, false, true, true, true, true, true, true, false], "max": [0.2899223864078522, 0.10584467649459839, 1.2854759693145752, 0.28685441613197327, 0.052155349403619766, 0.02761165425181389, 1.0532722473144531, 0.08897088468074799, 0.41264083981513977, 1.211844801902771, 0.2546408176422119, 0.17794176936149597, 0.1871456652879715, 1.0135692358016968], "mean": [0.08488845080137253, -0.3301914930343628, 1.0135648250579834, 0.01947842352092266, -0.2516394555568695, 0.004541649017482996, 0.6025070548057556, -0.11463526636362076, -0.09523569047451019, 0.8199716806411743, -0.027128169313073158, -0.190309077501297, 0.01706995628774166, 0.5188212394714355], "min": [-0.052155349403619766, -0.9464661478996277, 0.630466103553772, -0.1564660519361496, -0.5890486240386963, -0.0076699042692780495, -0.11616255342960358, -0.3021942377090454, -0.9433982372283936, 0.374291330575943, -0.23623304069042206, -0.5399612784385681, -0.01840776950120926, -0.05299859866499901], "p01": [-0.02454369328916073, -0.9035146832466125, 0.7163690328598022, -0.11811652034521103, -0.5387340784072876, -0.004601942375302315, -0.06382673233747482, -0.26691266894340515, -0.9081166386604309, 0.48166999220848083, -0.18285052478313446, -0.43718454241752625, -0.010737866163253784, -0.0078815296292305], "p99": [0.24083499610424042, 0.03528155758976936, 1.2363885641098022, 0.19634954631328583, -0.004601942375302315, 0.023316508159041405, 1.0298113822937012, 0.04141748324036598, 0.28685441613197327, 1.179631233215332, 0.1871456652879715, 0.1337631344795227, 0.18254372477531433, 0.9684521555900574], "std": [0.062149710953235626, 0.2356911301612854, 0.13307376205921173, 0.06283757835626602, 0.13161775469779968, 0.004605415277183056, 0.3009023070335388, 0.0619950145483017, 0.25287559628486633, 0.16802218556404114, 0.07270802557468414, 0.12436924129724503, 0.03112170472741127, 0.21067041158676147]}, "num_trajectories": 97, "num_transitions": 20881, "proprio_bimanual": {"max": [0.2914563715457916, 0.11351457983255386, 1.2962137460708618, 0.28378644585609436, 0.05522330850362778, 0.029145635664463043, 0.9999999403953552, 0.08283496648073196, 0.42951464653015137, 1.2225826978683472, 0.2546408176422119, 0.18100973963737488, 0.19174760580062866, 0.999051570892334], "mean": [0.07952567934989929, -0.33795416355133057, 1.0339573621749878, 0.01684681326150894, -0.2489086389541626, 0.004585095215588808, 0.6542050838470459, -0.11001477390527725, -0.0882151797413826, 0.8388670682907104, -0.027926674112677574, -0.18644249439239502, 0.01662961021065712, 0.5871848464012146], "min": [-0.04908738657832146, -0.9679418802261353, 0.6565437912940979, -0.1533980816602707, -0.5798447728157043, -0.0076699042692780495, 0.11382552981376648, -0.2960582971572876, -0.9725438356399536, 0.3880971372127533, -0.23623304069042206, -0.4770680367946625, -0.01840776950120926, 0.2971571683883667], "p01": [-0.02147573232650757, -0.9618059992790222, 0.7363107800483704, -0.11811652034521103, -0.5276893973350525, -0.004601942375302315, 0.11705803871154785, -0.2623107135295868, -0.9664078950881958, 0.49854376912117004, -0.18100973963737488, -0.41724279522895813, -0.00920388475060463, 0.30620047450065613], "p99": [0.2316311001777649, 0.04448544234037399, 1.2486604452133179, 0.19481556117534637, -0.004601942375302315, 0.02454369328916073, 0.9998724460601807, 0.03834952041506767, 0.30066025257110596, 1.1919031143188477, 0.1840776950120926, 0.1306951642036438, 0.18100973963737488, 0.9899495244026184], "std": [0.06084473431110382, 0.255412220954895, 0.13424108922481537, 0.06108736991882324, 0.1278107613325119, 0.0049295248463749886, 0.2951796054840088, 0.06230009347200394, 0.2886359989643097, 0.17274509370326996, 0.07182013243436813, 0.121708944439888, 0.03072475455701351, 0.18272094428539276]}}, "aloha_pick_place_dataset": {"action": {"mask": [true, true, true, true, true, true, false, true, true, true, true, true, true, false], "max": [0.401902973651886, -0.7900001406669617, 1.4173983335494995, 0.15033012628555298, 0.11658254265785217, 0.02454369328916073, 0.7573042511940002, 0.754718542098999, 1.1259418725967407, 1.4879614114761353, 1.6275537014007568, 1.7594760656356812, 2.22580623626709, 1.1254596710205078], "mean": [0.07092317938804626, -0.9651168584823608, 1.2104730606079102, -0.008339351043105125, -0.2655200660228729, 0.0025311745703220367, 0.025376606732606888, -0.08909047394990921, -0.16390836238861084, 0.6889074444770813, -0.01147516630589962, -0.013750928454101086, 0.19231532514095306, 0.7674521207809448], "min": [-0.03834952041506767, -1.1535536050796509, 0.9863497018814087, -0.22396120429039001, -0.5062136650085449, -0.03988350182771683, -0.08367816358804703, -1.0354371070861816, -1.199573040008545, -2.1168935298919678, -1.653631329536438, -0.9694758653640747, -1.0860583782196045, -0.07104548066854477], "p01": [-0.003067961661145091, -1.0860583782196045, 1.1090681552886963, -0.12425244599580765, -0.424912691116333, -0.004601942375302315, -0.04217046871781349, -0.5859806537628174, -1.0783885717391968, -0.2929903268814087, -1.023165225982666, -0.7286409139633179, -0.6350680589675903, 0.011970043182373047], "p99": [0.2377670258283615, -0.8513593673706055, 1.3222914934158325, 0.08130098134279251, 0.013805827125906944, 0.012271846644580364, 0.25560227036476135, 0.33594179153442383, 0.5936505794525146, 1.2547962665557861, 0.6979612708091736, 0.7654564380645752, 1.4818254709243774, 1.06229567527771], "std": [0.053478438407182693, 0.036589834839105606, 0.04218200966715813, 0.03957560285925865, 0.09045778214931488, 0.0035138705279678106, 0.07376480847597122, 0.20179739594459534, 0.4255509078502655, 0.4108298122882843, 0.2871744632720947, 0.34284743666648865, 0.3679707646369934, 0.28391531109809875]}, "num_trajectories": 2589, "num_transitions": 260626, "proprio_bimanual": {"max": [0.39730104804039, -0.7915341258049011, 1.4250681400299072, 0.14879614114761353, 0.11965050548315048, 0.026077674701809883, 0.8500267267227173, 0.7531846165657043, 1.0308351516723633, 1.4925633668899536, 1.6260197162628174, 1.716524600982666, 2.222738265991211, 0.9999999403953552], "mean": [0.06795695424079895, -0.9682918190956116, 1.2251989841461182, -0.008771675638854504, -0.25891485810279846, 0.001853941474109888, 0.1313188672065735, -0.08673743903636932, -0.1503143459558487, 0.7099950313568115, -0.00993292685598135, -0.007886354811489582, 0.19005371630191803, 0.8438357710838318], "min": [-0.03528155758976936, -1.1566215753555298, 1.0170292854309082, -0.21475732326507568, -0.4970097839832306, -0.03988350182771683, 0.11901140958070755, -1.0339031219482422, -1.2057089805603027, -1.7103886604309082, -1.6474953889846802, -0.9602720141410828, -1.084524393081665, 0.31624835729599], "p01": [-0.003067961661145091, -1.084524393081665, 1.1320778131484985, -0.12425244599580765, -0.4157088100910187, -0.004601942375302315, 0.12097528576850891, -0.5829127430915833, -1.084524393081665, -0.2653786838054657, -1.009359359741211, -0.7148350477218628, -0.6258642077445984, 0.32732486724853516], "p99": [0.2346990704536438, -0.8528933525085449, 1.3330293893814087, 0.0782330259680748, 0.019941750913858414, 0.010737866163253784, 0.21283377707004547, 0.33594179153442383, 0.6335340738296509, 1.2655341625213623, 0.6948933005332947, 0.7654564380645752, 1.4649516344070435, 0.9999957084655762], "std": [0.05304940789937973, 0.035672854632139206, 0.04050517454743385, 0.03849741443991661, 0.08970217406749725, 0.0032832431606948376, 0.043496303260326385, 0.20116861164569855, 0.4448918104171753, 0.4096622169017792, 0.2828507423400879, 0.33823955059051514, 0.3658812642097473, 0.21204698085784912]}}, "aloha_static_dataset": {"action": {"mask": [true, true, true, true, true, true, false, true, true, true, true, true, true, false], "max": [0.6166602969169617, 0.49854376912117004, 1.5339808464050293, 2.0447964668273926, 1.7149906158447266, 1.9696314334869385, 1.0695143938064575, 0.9065826535224915, 0.6366020441055298, 1.4572818279266357, 1.90827214717865, 2.057068347930908, 2.494252920150757, 1.1019988059997559], "mean": [0.03198118507862091, -0.326416015625, 0.882464587688446, 0.12333173304796219, -0.2511795163154602, -0.1770600825548172, 0.21229824423789978, -0.10698918998241425, -0.15385441482067108, 0.7678719758987427, -0.012271172367036343, -0.15685836970806122, 0.14149953424930573, 0.24556414783000946], "min": [-3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -2.4759624004364014, -0.8927768468856812, -1.199573040008545, -0.3221359848976135, -2.2288742065429688, -1.699650764465332, -1.886796474456787, -0.270430326461792], "p01": [-0.2899223864078522, -0.9940195679664612, 0.14112623035907745, -0.6181942820549011, -1.4373400211334229, -1.6321556568145752, -0.20208077132701874, -0.7056311964988708, -1.020097255706787, 0.019941750913858414, -1.7042527198791504, -1.2900779247283936, -1.274738073348999, -0.2141173779964447], "p99": [0.4801360070705414, 0.38196122646331787, 1.2870099544525146, 1.7410682439804077, 1.494097352027893, 1.072252631187439, 0.8913554549217224, 0.4801360070705414, 0.47093212604522705, 1.2870099544525146, 0.9188545346260071, 1.41586434841156, 2.104621648788452, 0.9864989519119263], "std": [0.1508723348379135, 0.39191514253616333, 0.2704477906227112, 0.4883921444416046, 0.6031206846237183, 0.46172595024108887, 0.3162362575531006, 0.22893957793712616, 0.340520977973938, 0.28399741649627686, 0.43075162172317505, 0.6010551452636719, 0.6239916086196899, 0.3356647193431854]}, "num_trajectories": 855, "num_transitions": 513850, "proprio_bimanual": {"max": [0.5967185497283936, 0.5154175758361816, 1.546252727508545, 2.0417284965515137, 1.7073206901550293, 1.968097448348999, 0.9999999403953552, 0.898912787437439, 0.6580777764320374, 1.4695535898208618, 1.8960002660751343, 2.0447964668273926, 2.491184949874878, 0.9999999403953552], "mean": [0.030769599601626396, -0.3214639127254486, 0.8988732695579529, 0.12243737280368805, -0.24422065913677216, -0.17665816843509674, 0.23732569813728333, -0.105474554002285, -0.14326737821102142, 0.787182092666626, -0.011778720654547215, -0.1505511999130249, 0.1406756043434143, 0.28380194306373596], "min": [-0.6028544902801514, -1.1489516496658325, -0.13499031960964203, -2.2411460876464844, -1.5155730247497559, -2.0678062438964844, -0.1960025131702423, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -3.1415927410125732, -0.1764688491821289], "p01": [-0.28685441613197327, -0.9986215233802795, 0.1564660519361496, -0.6151263117790222, -1.4296700954437256, -1.6336896419525146, -0.15736174583435059, -0.7025632262229919, -1.026233196258545, 0.04448544234037399, -1.6950488090515137, -1.2808740139007568, -1.2762720584869385, -0.16005246341228485], "p99": [0.4770680367946625, 0.401902973651886, 1.3008157014846802, 1.73493230342865, 1.4879614114761353, 1.0615147352218628, 0.9563239216804504, 0.47860202193260193, 0.49547579884529114, 1.3023496866226196, 0.9142525792121887, 1.4112623929977417, 2.104621648788452, 0.9946948885917664], "std": [0.14927177131175995, 0.403738409280777, 0.2707565128803253, 0.48628538846969604, 0.5983042120933533, 0.4609971046447754, 0.33456477522850037, 0.22781017422676086, 0.35364189743995667, 0.28373369574546814, 0.4273151159286499, 0.5964325070381165, 0.6225544810295105, 0.3423353433609009]}}, "aloha_sushi_cut_full_dataset": {"action": {"mask": [true, true, true, true, true, true, false, true, true, true, true, true, true, false], "max": [0.6320000886917114, 0.4279806613922119, 1.4465439319610596, 0.41724279522895813, 0.6580777764320374, 0.7455146908760071, 1.0659050941467285, 0.7792622447013855, 1.087592363357544, 1.38825261592865, 2.552544116973877, 1.9113401174545288, 3.103243112564087, 1.1182408332824707], "mean": [-0.02789231389760971, -0.2545304000377655, 0.9634057879447937, -0.11814907938241959, -0.365222305059433, 0.04809948801994324, 0.45852527022361755, -0.050104331225156784, -0.1734812706708908, 0.6589794754981995, -0.06797567009925842, 0.43668055534362793, 0.33219799399375916, 0.5356466174125671], "min": [-0.5123496055603027, -1.0339031219482422, 0.374291330575943, -0.8421554565429688, -1.0477088689804077, -1.6919808387756348, -0.09811566770076752, -0.9710098505020142, -1.115204095840454, -2.2380781173706055, -3.130854845046997, -1.6352235078811646, -1.0615147352218628, -0.07645954936742783], "p01": [-0.3267379105091095, -0.9480001330375671, 0.5123496055603027, -0.7025632262229919, -0.9464661478996277, -0.015339808538556099, -0.03314702585339546, -0.7240389585494995, -0.8805050253868103, -0.2960582971572876, -1.7824857234954834, -0.37275734543800354, -0.5829127430915833, 0.02821209840476513], "p99": [0.40343695878982544, 0.2853204309940338, 1.3422331809997559, 0.18867963552474976, 0.2162145972251892, 0.5568350553512573, 1.0207879543304443, 0.4141748249530792, 0.3988350033760071, 1.1581555604934692, 0.7010292410850525, 1.5677284002304077, 2.287165403366089, 1.038834810256958], "std": [0.15365876257419586, 0.36534032225608826, 0.22074617445468903, 0.1681865155696869, 0.24236762523651123, 0.1319788545370102, 0.3341032862663269, 0.27787142992019653, 0.27358198165893555, 0.34551018476486206, 0.46538832783699036, 0.42368343472480774, 0.6024542450904846, 0.281799852848053]}, "num_trajectories": 1081, "num_transitions": 476206, "proprio_bimanual": {"max": [0.6289321184158325, 0.3942330777645111, 1.4603497982025146, 0.4141748249530792, 0.647339940071106, 0.7439807057380676, 0.9999999403953552, 0.7761942744255066, 0.8958448171615601, 1.3974565267562866, 2.549476146697998, 1.9098061323165894, 3.0986413955688477, 0.9999999403953552], "mean": [-0.027425872161984444, -0.25034773349761963, 0.9803621172904968, -0.11842118203639984, -0.3597453236579895, 0.04707001894712448, 0.5113943219184875, -0.04751736298203468, -0.15449005365371704, 0.676561176776886, -0.06781305372714996, 0.4391832649707794, 0.3313209116458893, 0.6269756555557251], "min": [-0.5154175758361816, -1.0354371070861816, 0.4049709439277649, -0.8421554565429688, -1.0246992111206055, -1.6950488090515137, 0.11318255215883255, -0.9664078950881958, -1.121340036392212, -1.6935148239135742, -3.1216509342193604, -1.6306216716766357, -1.0599807500839233, 0.2837609052658081], "p01": [-0.31753402948379517, -0.954136073589325, 0.5322913527488708, -0.7040972113609314, -0.9357283115386963, -0.016873789951205254, 0.11705803871154785, -0.7225049734115601, -0.9081166386604309, -0.2761165499687195, -1.7840197086334229, -0.36201947927474976, -0.5813787579536438, 0.3107548654079437], "p99": [0.40343695878982544, 0.2822524607181549, 1.3606410026550293, 0.18561168015003204, 0.21322333812713623, 0.5568350553512573, 0.9993929266929626, 0.4141748249530792, 0.43104860186576843, 1.1765632629394531, 0.6979612708091736, 1.5677284002304077, 2.287165403366089, 0.9999483823776245], "std": [0.15067370235919952, 0.37187567353248596, 0.22080998122692108, 0.16786710917949677, 0.24047496914863586, 0.13174571096897125, 0.3340302109718323, 0.27747249603271484, 0.2827145755290985, 0.3439193367958069, 0.46378853917121887, 0.42283692955970764, 0.6015809178352356, 0.2582082450389862]}}, "austin_buds_dataset_converted_externally_to_rlds": {"action": {"mask": [true, true, true, true, true, true, false], "max": [1.0, 1.0, 1.0, 0.0, 0.0, 0.0, 1.0], "mean": [-0.07678326219320297, 0.0036849179305136204, 0.05644919350743294, 0.0, 0.0, 0.0, 0.3510494828224182], "min": [-1.0, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0], "p01": [-1.0, -0.9599999785423279, -0.8714285492897034, 0.0, 0.0, 0.0, 0.0], "p99": [1.0, 0.8600000143051147, 1.0, 0.0, 0.0, 0.0, 1.0], "std": [0.6367745399475098, 0.37889161705970764, 0.47796347737312317, 0.0, 0.0, 0.0, 0.47721067070961]}, "num_trajectories": 50, "num_transitions": 34112}, "austin_sailor_dataset_converted_externally_to_rlds": {"action": {"mask": [true, true, true, true, true, true, false], "max": [1.0, 1.0, 1.0, 0.0, 0.0, 0.375, 1.0], "mean": [0.011825432069599628, 0.006461071316152811, 0.0602363757789135, 0.0, 0.0, 0.0016465900698676705, 0.5260950326919556], "min": [-1.0, -1.0, -1.0, 0.0, 0.0, -0.375, 0.0], "p01": [-1.0, -0.9828571677207947, -0.6000000238418579, 0.0, 0.0, -0.17249999940395355, 0.0], "p99": [1.0, 0.9457142949104309, 1.0, 0.0, 0.0, 0.17892856895923615, 1.0], "std": [0.46348774433135986, 0.4124029278755188, 0.41186290979385376, 0.0, 0.0, 0.05786094069480896, 0.4989347755908966]}, "num_trajectories": 240, "num_transitions": 353094}, "austin_sirius_dataset_converted_externally_to_rlds": {"action": {"mask": [true, true, true, true, true, true, false], "max": [1.0002285242080688, 0.960608720779419, 1.105179786682129, 0.0, 0.0, 0.341785728931427, 1.0], "mean": [0.07747671753168106, 0.03195551782846451, 0.04244737699627876, 0.0, 0.0, -0.016034552827477455, 0.43260180950164795], "min": [-1.0183025598526, -0.9800000190734863, -0.9774575233459473, 0.0, 0.0, -0.34607142210006714, 0.0], "p01": [-0.7809058427810669, -0.5667179226875305, -0.5254343152046204, 0.0, 0.0, -0.2849509119987488, 0.0], "p99": [0.9569637775421143, 0.6971374750137329, 0.8124887943267822, 0.0, 0.0, 0.1971428543329239, 1.0], "std": [0.39063239097595215, 0.29981499910354614, 0.27822697162628174, 0.0, 0.0, 0.08120565116405487, 0.49528470635414124]}, "num_trajectories": 559, "num_transitions": 279939}, "bc_z": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.2165454924106598, 0.1251407265663147, 0.10772687941789627, 0.33544227480888367, 0.28117990493774414, 0.40614867210388184, 1.0], "mean": [-0.009957773610949516, 0.0008958547259680927, 0.0049958485178649426, 0.00029754548450000584, -0.008734275586903095, -0.03069145977497101, 0.8344562649726868], "min": [-0.1677047461271286, -0.14630407094955444, -0.10066790133714676, -0.29421567916870117, -0.32101404666900635, -0.4635624885559082, 0.0], "p01": [-0.09220655262470245, -0.06456145644187927, -0.049121275544166565, -0.11594625562429428, -0.1415254920721054, -0.2251061648130417, 0.0], "p99": [0.07628867030143738, 0.0580197349190712, 0.052540797740221024, 0.11740604043006897, 0.117039754986763, 0.16729307174682617, 1.0], "std": [0.030538640916347504, 0.023141583427786827, 0.020642967894673347, 0.04155871644616127, 0.046428218483924866, 0.0769806057214737, 0.36104875802993774]}, "num_trajectories": 43264, "num_transitions": 6015535}, "berkeley_autolab_ur5": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.019999999552965164, 0.019999999552965164, 0.019999999552965164, 0.06666667014360428, 0.06666667014360428, 0.06666667014360428, 1.0], "mean": [0.0005683630588464439, 0.0012176920427009463, -0.0005296376184560359, 0.00021029730851296335, 6.069445953471586e-05, 0.0012049878714606166, 0.6298308372497559], "min": [-0.019999999552965164, -0.019999999552965164, -0.019999999552965164, -0.06666667014360428, -0.06666667014360428, -0.06666667014360428, 0.0], "p01": [-0.019999999552965164, -0.019999999552965164, -0.019999999552965164, -0.02628571353852749, -0.06666667014360428, -0.03847619146108627, 0.0], "p99": [0.019999999552965164, 0.019999999552965164, 0.019999999552965164, 0.031809523701667786, 0.06666667014360428, 0.036571428179740906, 1.0], "std": [0.011533336713910103, 0.007990485988557339, 0.009577815420925617, 0.009433039464056492, 0.016427572816610336, 0.011054139584302902, 0.48267897963523865]}, "num_trajectories": 1000, "num_transitions": 97939}, "berkeley_cable_routing": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.9633283019065857, 1.0, 1.0, 0.0, 0.0, 1.0, 0.0], "mean": [-0.07139885425567627, 0.023609021678566933, 0.10241945087909698, 0.0, 0.0, 0.049670953303575516, 0.0], "min": [-0.9809081554412842, -0.9554349184036255, -0.9994775056838989, 0.0, 0.0, -1.0, 0.0], "p01": [-0.5534318685531616, -0.4797285497188568, -0.5314934849739075, 0.0, 0.0, -0.8855219483375549, 0.0], "p99": [0.42652836441993713, 0.5000944137573242, 0.6398234367370605, 0.0, 0.0, 0.9842438697814941, 0.0], "std": [0.181549534201622, 0.18109826743602753, 0.2122078388929367, 0.0, 0.0, 0.34755080938339233, 0.0]}, "num_trajectories": 1647, "num_transitions": 42328}, "berkeley_fanuc_manipulation": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.009999999776482582, 0.009999999776482582, 0.009999999776482582, 0.03490658476948738, 0.03490658476948738, 0.03490658476948738, 1.0], "mean": [0.0007744057802483439, -0.00031240080716088414, -0.0015001941937953234, -0.0007515158504247665, -0.00015832878125365824, 0.00014327642566058785, 0.699295699596405], "min": [-0.009999999776482582, -0.009999999776482582, -0.009999999776482582, -0.03490658476948738, -0.03490658476948738, -0.03490658476948738, 0.0], "p01": [-0.009999999776482582, -0.009999999776482582, -0.009999999776482582, -0.03490658476948738, 0.0, -0.03490658476948738, 0.0], "p99": [0.009999999776482582, 0.009999999776482582, 0.009999999776482582, 0.03490658476948738, 0.0, 0.03490658476948738, 1.0], "std": [0.003407009644433856, 0.004992189351469278, 0.005344335921108723, 0.007598913740366697, 0.004081867169588804, 0.008568980731070042, 0.45869025588035583]}, "num_trajectories": 415, "num_transitions": 62613}, "bridge_dataset": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.41691166162490845, 0.25864794850349426, 0.21218234300613403, 3.122201919555664, 1.8618112802505493, 6.280478477478027, 1.0], "mean": [0.0002175865665776655, 0.00012508276267908514, -0.00017108430620282888, -0.00016170882736332715, -0.00025248737074434757, 0.0002515808155294508, 0.5879485607147217], "min": [-0.4007510244846344, -0.13874775171279907, -0.22553899884223938, -3.2010786533355713, -1.8618112802505493, -6.279075622558594, 0.0], "p01": [-0.02853955328464508, -0.04143204540014267, -0.02597738429903984, -0.08020886778831482, -0.09213060140609741, -0.20548619329929352, 0.0], "p99": [0.028122276067733765, 0.040630318224430084, 0.03994889184832573, 0.0812191590666771, 0.07724379003047943, 0.20214049518108368, 1.0], "std": [0.009632375091314316, 0.013500677421689034, 0.012510561384260654, 0.028145212680101395, 0.03028246760368347, 0.07585537433624268, 0.4877188801765442]}, "num_trajectories": 60064, "num_transitions": 2195527}, "cmu_stretch": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.02338407188653946, 0.0, 0.023404927924275398, 0.0, 0.0, 0.0, 1.0], "mean": [0.0003630469145718962, 0.0, 0.0016466975212097168, 0.0, 0.0, 0.0, 0.3987048268318176], "min": [-0.019353797659277916, 0.0, -0.02019215188920498, 0.0, 0.0, 0.0, 0.0], "p01": [-0.011175686493515968, 0.0, -0.003220636397600174, 0.0, 0.0, 0.0, 0.0], "p99": [0.014501785859465599, 0.0, 0.015056106261909008, 0.0, 0.0, 0.0, 1.0], "std": [0.0040818192064762115, 0.0, 0.003774337936192751, 0.0, 0.0, 0.0, 0.489637166261673]}, "num_trajectories": 135, "num_transitions": 25016}, "dlr_edan_shared_control_converted_externally_to_rlds": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.18991442024707794, 0.0739002525806427, 0.18064819276332855, 0.0866486132144928, 0.13464981317520142, 0.16910280287265778, 1.0], "mean": [0.006647807080298662, -0.0007657374371774495, 0.006522843614220619, 0.001167974784038961, -0.006395628675818443, -0.01190300565212965, 0.6985887289047241], "min": [-0.10054297000169754, -0.08427435159683228, -0.13533438742160797, -0.17556548118591309, -0.18485672771930695, -0.2680685818195343, 0.0], "p01": [-0.029871223494410515, -0.06013263016939163, -0.08286409080028534, -0.059244442731142044, -0.15986867249011993, -0.15636983513832092, 0.0], "p99": [0.08832091838121414, 0.042126186192035675, 0.11311905831098557, 0.0643695592880249, 0.03941855952143669, 0.15664684772491455, 1.0], "std": [0.021393612027168274, 0.018142299726605415, 0.03374375402927399, 0.017435431480407715, 0.03394375741481781, 0.04641878232359886, 0.45885777473449707]}, "num_trajectories": 104, "num_transitions": 8928}, "droid": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.9999998211860657, 0.999991774559021, 0.9999973177909851, 0.9999874830245972, 0.9999954104423523, 0.9999998807907104, 1.0], "mean": [0.024281324818730354, -0.0028759415727108717, 0.018233438953757286, 0.0038624575827270746, -0.03020552359521389, -0.007724810857325792, 0.42078232765197754], "min": [-0.9999999403953552, -0.9999951124191284, -0.9999960660934448, -0.9999980330467224, -0.9999979734420776, -0.9999998807907104, 0.0], "p01": [-0.7671820521354675, -0.5812531113624573, -0.5680612921714783, -0.6462401151657104, -0.6986615657806396, -0.8940888047218323, 0.0], "p99": [0.7328272461891174, 0.5705102682113647, 0.734496533870697, 0.6701276302337646, 0.6458622217178345, 0.8935507535934448, 1.0], "std": [0.2521198093891144, 0.18553116917610168, 0.22702498733997345, 0.21980907022953033, 0.22736462950706482, 0.2911641299724579, 0.4378896951675415]}, "num_trajectories": 76500, "num_transitions": 23129514, "proprio_bimanual": {"max": [0.8575563430786133, 0.8196876049041748, 1.0111403465270996, 3.1415927410125732, 1.5705928802490234, 3.1415927410125732, 1.0], "mean": [0.5301907658576965, 0.0006110519752837718, 0.3207733631134033, 0.3172304332256317, -0.09120110422372818, -0.05302700027823448, 0.3795362710952759], "min": [-0.22192564606666565, -0.827903687953949, -0.24001094698905945, -3.141592502593994, -1.5703768730163574, -3.141563653945923, 0.0], "p01": [0.26736223697662354, -0.4401699900627136, -0.042780909687280655, -3.1375010013580322, -1.2102818489074707, -2.137446403503418, 0.0], "p99": [0.7802324891090393, 0.44093888998031616, 0.7827594876289368, 3.137559413909912, 0.8736637234687805, 2.017641544342041, 0.9911894202232361], "std": [0.11526807397603989, 0.1750902533531189, 0.16158035397529602, 2.781348466873169, 0.3448600471019745, 0.7497172355651855, 0.41822928190231323]}}, "droid_wipe": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.9669402837753296, 0.9781726002693176, 0.4805821478366852, 0.3767365515232086, 0.31902551651000977, 0.5357517004013062, 1.0], "mean": [0.03723526746034622, 0.03707795962691307, -0.011916602961719036, 0.02478920854628086, -0.0015099693555384874, 0.003385246032848954, 0.4734032154083252], "min": [-0.8601731061935425, -0.34971675276756287, -0.626416027545929, -0.299400269985199, -0.3550586402416229, -0.5534208416938782, 0.0], "p01": [-0.428209513425827, -0.23935642838478088, -0.39668455719947815, -0.16798557341098785, -0.19491541385650635, -0.3109041154384613, 0.009911894798278809], "p99": [0.4129367768764496, 0.36322423815727234, 0.362138956785202, 0.2562437951564789, 0.20809145271778107, 0.35650333762168884, 1.0], "std": [0.18475475907325745, 0.14856348931789398, 0.15049704909324646, 0.09076415002346039, 0.08990494906902313, 0.12920668721199036, 0.4628407955169678]}, "num_trajectories": 70, "num_transitions": 26728, "proprio_bimanual": {"max": [0.7966542840003967, 0.33264991641044617, 0.5719268321990967, 3.1415889263153076, 0.36567333340644836, 1.0877469778060913, 0.9603524208068848], "mean": [0.6117352247238159, 0.056952882558107376, 0.20608071982860565, 1.9908137321472168, 0.11739496141672134, 0.12222489714622498, 0.44524455070495605], "min": [0.26999974250793457, -0.1950942873954773, 0.1212514117360115, -3.1415605545043945, -0.2757776379585266, -0.3994779586791992, 0.0], "p01": [0.31304043531417847, -0.15182510018348694, 0.12411060929298401, -3.137579917907715, -0.14648257195949554, -0.35721486806869507, 0.0], "p99": [0.7794656753540039, 0.3107130825519562, 0.5103863477706909, 3.137955904006958, 0.3189685046672821, 0.6602705121040344, 0.9603524208068848], "std": [0.10104220360517502, 0.14455406367778778, 0.09893947839736938, 2.311044692993164, 0.0939084067940712, 0.20109158754348755, 0.3899177312850952]}}, "fractal20220817_data": {"action": {"mask": [true, true, true, true, true, true, false], "max": [2.9984593391418457, 22.09052848815918, 2.7507524490356445, 1.570636510848999, 1.5321086645126343, 1.5691522359848022, 1.0], "mean": [0.0069875395856797695, 0.006265789270401001, -0.012625272385776043, 0.043333787471055984, -0.005756153725087643, 0.0009130391990765929, 0.5354204773902893], "min": [-2.0204520225524902, -5.497899532318115, -2.031663417816162, -1.569917917251587, -1.569892168045044, -1.570419430732727, 0.0], "p01": [-0.224535271525383, -0.1482001394033432, -0.23158970475196838, -0.35179948806762695, -0.4193011224269867, -0.4364345967769623, 0.0], "p99": [0.17824687063694, 0.1493837982416153, 0.21842354536056519, 0.5892665982246399, 0.3527265787124634, 0.4479668140411377, 1.0], "std": [0.06921233236789703, 0.05970727652311325, 0.07353049516677856, 0.1561039537191391, 0.1316443383693695, 0.14593897759914398, 0.49711793661117554]}, "num_trajectories": 87212, "num_transitions": 3786400}, "furniture_bench_dataset_converted_externally_to_rlds": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.10000000149011612, 0.10000000149011612, 0.10000000149011612, 0.8651833534240723, 1.0909736156463623, 2.863185405731201, 1.0], "mean": [0.0001461088832002133, 0.001083097537048161, 0.0006224964745342731, -0.003303213743492961, -0.002688063308596611, 0.018242694437503815, 0.48854944109916687], "min": [-0.10495579987764359, -0.10939455777406693, -0.10000000149011612, -0.971906840801239, -1.0475432872772217, -3.06000018119812, 0.0], "p01": [-0.053988199681043625, -0.05049169436097145, -0.03249924257397652, -0.19538870453834534, -0.4167456030845642, -0.8886768221855164, 0.0], "p99": [0.054148413240909576, 0.049651648849248886, 0.060055799782276154, 0.18231667578220367, 0.39867785573005676, 0.8772024512290955, 1.0], "std": [0.016107231378555298, 0.014891562983393669, 0.014014188200235367, 0.05827421322464943, 0.11417066305875778, 0.334794819355011, 0.4999154210090637]}, "num_trajectories": 5100, "num_transitions": 3948057}, "go1": {"action": {"mask": [true, true, true, true, true, true, true, true, true, true, true, true], "max": [0.9711780548095703, 0.994792103767395, 0.9837396144866943, 0.9897138476371765, 0.9882936477661133, 0.9864842891693115, 0.9902452826499939, 0.9898523688316345, 0.9920899868011475, 0.9851218461990356, 0.9933716058731079, 0.9971944689750671], "mean": [-0.43288278579711914, 0.13856956362724304, -0.052038900554180145, 0.41994938254356384, 0.321550190448761, -0.13532669842243195, -0.13703575730323792, 0.537114679813385, 0.30219823122024536, 0.07318709790706635, 0.5988545417785645, 0.35771679878234863], "min": [-0.9793693423271179, -0.9754825830459595, -0.987050473690033, -0.9412225484848022, -0.9835582971572876, -0.9946646690368652, -0.997287392616272, -0.9939435124397278, -0.9923244118690491, -0.9963865280151367, -0.9858922958374023, -0.9982498288154602], "p01": [-0.9504456520080566, -0.9573942422866821, -0.9764307141304016, -0.8477075099945068, -0.8979476094245911, -0.9801793694496155, -0.975758969783783, -0.8001573085784912, -0.9776679873466492, -0.9768717288970947, -0.9437603950500488, -0.9882175326347351], "p99": [0.7414185404777527, 0.9827137589454651, 0.9642788171768188, 0.9754635691642761, 0.9787585139274597, 0.965583860874176, 0.9755379557609558, 0.973909854888916, 0.9703678488731384, 0.9696959853172302, 0.9869292974472046, 0.9755439162254333], "std": [0.48112753033638, 0.8677573204040527, 0.7942482829093933, 0.6009141206741333, 0.7250922918319702, 0.8250295519828796, 0.7768806219100952, 0.49424421787261963, 0.6407467722892761, 0.8372606635093689, 0.6176847815513611, 0.7633413672447205]}, "num_trajectories": 100, "num_transitions": 30000, "proprio_quadruped": {"max": [1.7235063314437866, 2.008730888366699, 1.688401460647583, 1.340979814529419, 0.22817234694957733, 0.23849895596504211, 2.796604871749878, 1.3915307521820068, -0.0027559020090848207, 1.4903150796890259, -1.4378445148468018, 0.6652379035949707, 1.3966472148895264, -1.5677262544631958, 0.14248280227184296, 1.7315963506698608, -1.1821818351745605, 0.3855018615722656, 1.768029808998108, -1.2137691974639893, 1.9728423357009888, 3.1757376194000244, 2.5918877124786377, 2.5459682941436768, 3.3386757373809814, 1.9510498046875, 3.1337480545043945, 3.689176082611084, 2.20379376411438, 2.8383073806762695, 2.9759681224823, 1.8330326080322266, 0.9705122709274292, 0.6030824184417725, 0.3648141324520111, 8.835007667541504, 16.44219398498535, 15.395484924316406, 4.007290363311768, 12.887717247009277, 18.893701553344727, 9.359314918518066, 7.124743938446045, 9.38571548461914, 8.388808250427246, 12.165282249450684, 11.602814674377441, 0.2898794412612915, 1.4142695665359497, 0.0, 0.42804068326950073, 1.4088549613952637, 0.0, 0.2975340187549591, 1.5408639907836914, 0.0, 0.42651864886283875, 1.54354989528656, 0.0], "mean": [0.6831021904945374, 0.7269537448883057, 0.4211779534816742, 0.49006208777427673, -0.03447343036532402, 0.05436777323484421, 0.007375824730843306, -0.025733627378940582, -0.33010342717170715, 0.8269442319869995, -1.8909451961517334, 0.34065914154052734, 0.8792008757591248, -1.9987857341766357, -0.1800752580165863, 1.4174343347549438, -1.4719735383987427, 0.13358385860919952, 1.322582483291626, -1.6205711364746094, -0.007464180700480938, -0.004801904782652855, -0.01328329835087061, -0.00010208530147792771, 0.04918842762708664, -0.01929856464266777, -0.013528216630220413, 0.0374407060444355, 0.017683126032352448, 0.023636464029550552, 0.04683077335357666, 0.014575446955859661, 0.4630736708641052, 0.031608086079359055, 0.01510164886713028, 1.8851922750473022, 0.29249051213264465, 5.011102199554443, -2.234726667404175, 0.9618292450904846, 6.5630974769592285, 1.1023426055908203, -4.277846813201904, -1.0017342567443848, -0.772983729839325, -1.153272271156311, 2.124343156814575, -0.23630324006080627, 0.8396673202514648, -1.6310995817184448, 0.22898787260055542, 0.9287782311439514, -1.664103388786316, -0.12316616624593735, 1.2048838138580322, -1.5083084106445312, 0.09686200320720673, 1.266514778137207, -1.5012751817703247], "min": [0.0, 0.0, 0.0, 0.0, -0.37570881843566895, -0.2607632875442505, -2.194438934326172, -1.5304193496704102, -0.5564890503883362, 0.25805091857910156, -2.308239459991455, -0.1646329164505005, 0.3551809787750244, -2.3836452960968018, -0.36382460594177246, 0.7306174039840698, -1.8021456003189087, -0.20499715209007263, 0.636256217956543, -1.8915094137191772, -2.3055882453918457, -5.796142101287842, -3.9501845836639404, -2.9260988235473633, -4.984919548034668, -3.9771015644073486, -2.398177146911621, -2.559072494506836, -2.439643621444702, -3.6575546264648438, -5.093038082122803, -2.6396498680114746, -0.12662868201732635, -0.6248123645782471, -0.6151577234268188, -2.6750080585479736, -20.100000381469727, -6.598437786102295, -11.888065338134766, -16.59612274169922, -4.1526360511779785, -6.920740604400635, -19.234848022460938, -9.754408836364746, -7.828747749328613, -10.395963668823242, -6.097836017608643, -0.42461198568344116, 0.0, -1.9681544303894043, -0.2778538167476654, 0.0, -1.971219778060913, -0.4305509030818939, 0.0, -1.9702775478363037, -0.2999993860721588, 0.0, -1.9726630449295044], "p01": [0.0, 0.0, 0.0, 0.0, -0.31789064407348633, -0.1611875593662262, -1.902925729751587, -1.0383254289627075, -0.4860880672931671, 0.3009587526321411, -2.2199134826660156, 0.03395313024520874, 0.3967101275920868, -2.3469905853271484, -0.33797919750213623, 0.7959285378456116, -1.7436975240707397, -0.14955174922943115, 0.7035114169120789, -1.86129891872406, -1.4765657186508179, -4.935683727264404, -2.2634594440460205, -2.53610897064209, -3.9845614433288574, -3.433659315109253, -2.0423455238342285, -1.9788025617599487, -2.1947786808013916, -2.2901523113250732, -4.591526031494141, -2.4168198108673096, 0.11962228268384933, -0.45827198028564453, -0.29232409596443176, -1.2787710428237915, -17.356660842895508, -4.718389511108398, -9.630339622497559, -12.325888633728027, -2.778402328491211, -4.709886074066162, -17.22472381591797, -8.89868450164795, -6.5760297775268555, -9.405510902404785, -5.301324367523193, -0.4150356352329254, 0.16531381011009216, -1.9638627767562866, -0.24017921090126038, 0.18489396572113037, -1.9653809070587158, -0.42339450120925903, 0.3324376344680786, -1.9642192125320435, -0.2922326922416687, 0.29928645491600037, -1.9684964418411255], "p99": [1.4877599477767944, 1.8338783979415894, 1.1704827547073364, 1.1214395761489868, 0.1849319338798523, 0.2016012966632843, 2.392596483230591, 0.7696436643600464, -0.10000000149011612, 1.4221614599227905, -1.5010769367218018, 0.5845762491226196, 1.3168748617172241, -1.689188838005066, 0.07910158485174179, 1.6856025457382202, -1.227604627609253, 0.3690935969352722, 1.741384506225586, -1.240753173828125, 1.5284068584442139, 2.6139142513275146, 2.3910694122314453, 1.3218952417373657, 2.9378936290740967, 1.6361223459243774, 2.603546142578125, 2.4264705181121826, 1.846211552619934, 2.1829051971435547, 2.615753412246704, 1.5877503156661987, 0.8173536062240601, 0.4705802798271179, 0.24554118514060974, 6.764741897583008, 14.570862770080566, 14.164921760559082, 1.8572927713394165, 11.548521041870117, 17.969152450561523, 8.055704116821289, 3.5118463039398193, 7.164917469024658, 7.244483947753906, 11.02392578125, 10.659111022949219, 0.1975473314523697, 1.4042056798934937, -1.3214612007141113, 0.4233253002166748, 1.4008913040161133, -1.320730209350586, 0.2916424870491028, 1.528680443763733, -1.319998860359192, 0.42138633131980896, 1.5386474132537842, -1.3191940784454346], "std": [0.4423843324184418, 0.5818858742713928, 0.3712891638278961, 0.3593256175518036, 0.14841414988040924, 0.07990722358226776, 1.145760416984558, 0.3990897238254547, 0.08310013264417648, 0.3483200669288635, 0.19282671809196472, 0.12398844957351685, 0.27255237102508545, 0.17192861437797546, 0.12384508550167084, 0.17459580302238464, 0.15298184752464294, 0.15351639688014984, 0.3299761414527893, 0.1862879991531372, 0.7029165625572205, 2.193232297897339, 1.3002904653549194, 0.9200616478919983, 1.7526696920394897, 1.265384316444397, 0.9295135140419006, 1.110544204711914, 1.0542253255844116, 1.0275239944458008, 2.0774941444396973, 1.1706202030181885, 0.1551482230424881, 0.2398366779088974, 0.12331152707338333, 2.055959463119507, 8.40924072265625, 6.041410446166992, 3.238286256790161, 5.919294357299805, 6.979445934295654, 3.4265213012695312, 5.518499374389648, 3.903205394744873, 3.967529535293579, 5.406657695770264, 5.190183639526367, 0.17088936269283295, 0.5608670711517334, 0.2965868413448334, 0.2151089608669281, 0.4964425265789032, 0.30576467514038086, 0.2847498655319214, 0.35308241844177246, 0.2398456186056137, 0.3046417832374573, 0.41500070691108704, 0.273499071598053]}}, "iamlab_cmu_pickup_insert_converted_externally_to_rlds": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.6634981632232666, 0.23428471386432648, 0.4308285415172577, 3.1415927410125732, 0.13647015392780304, 3.141592502593994, 1.0], "mean": [0.5274299383163452, 0.028582043945789337, 0.18712450563907623, 1.2339491844177246, 0.03226636350154877, -1.419951319694519, 0.5550631880760193], "min": [0.3071657121181488, -0.29754969477653503, 0.06578229367733002, -3.1415927410125732, -0.04584203287959099, -3.141592502593994, 0.0], "p01": [0.31488969922065735, -0.20317550003528595, 0.06785467267036438, -3.1409528255462646, -0.029743434861302376, -3.1410913467407227, 0.0], "p99": [0.6472805738449097, 0.2084680199623108, 0.36855655908584595, 3.1409926414489746, 0.11424950510263443, 3.141096830368042, 1.0], "std": [0.0810835063457489, 0.1116754338145256, 0.07747511565685272, 2.8737189769744873, 0.027747035026550293, 2.7678728103637695, 0.4969642460346222]}, "num_trajectories": 631, "num_transitions": 146241}, "jaco_play": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.20000000298023224, 0.20000000298023224, 0.20000000298023224, 0.0, 0.0, 0.0, 1.0], "mean": [0.0009658281342126429, -0.00580081669613719, -0.00395041611045599, 0.0, 0.0, 0.0, 0.34934908151626587], "min": [-0.20000000298023224, -0.20000000298023224, -0.20000000298023224, 0.0, 0.0, 0.0, 0.0], "p01": [-0.20000000298023224, -0.20000000298023224, -0.20000000298023224, 0.0, 0.0, 0.0, 0.0], "p99": [0.20000000298023224, 0.20000000298023224, 0.20000000298023224, 0.0, 0.0, 0.0, 1.0], "std": [0.12235039472579956, 0.09678776562213898, 0.11155413091182709, 0.0, 0.0, 0.0, 0.47682908177375793]}, "num_trajectories": 1085, "num_transitions": 77965}, "kuka": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.1697135865688324, 0.2777623236179352, 0.43710532784461975, 0.0, 0.0, 1.9684287309646606, 1.0], "mean": [-0.00046689563896507025, 0.000401388097088784, -0.0012808009050786495, 0.0, 0.0, -0.03722419589757919, 0.4131543040275574], "min": [-0.159867063164711, -0.2892282009124756, -0.2795473635196686, 0.0, 0.0, -1.9875637292861938, 0.0], "p01": [-0.06619441509246826, -0.08713878691196442, -0.15083016455173492, 0.0, 0.0, -0.541569709777832, 0.0], "p99": [0.06601838767528534, 0.08732476830482483, 0.1816817969083786, 0.0, 0.0, 0.2923380434513092, 1.0], "std": [0.02083253674209118, 0.029158594086766243, 0.06422988325357437, 0.0, 0.0, 0.1422533094882965, 0.49086299538612366]}, "num_trajectories": 209880, "num_transitions": 2455879}, "language_table": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.23357294499874115, 0.24496802687644958, 0.0, 0.0, 0.0, 0.0, 1.0], "mean": [0.0001489185815444216, -0.0005636646528728306, 0.0, 0.0, 0.0, 0.0, 1.0], "min": [-0.21989956498146057, -0.23736150562763214, 0.0, 0.0, 0.0, 0.0, 1.0], "p01": [-0.08179590106010437, -0.11795833706855774, 0.0, 0.0, 0.0, 0.0, 1.0], "p99": [0.08822273463010788, 0.11916933208703995, 0.0, 0.0, 0.0, 0.0, 1.0], "std": [0.030162891373038292, 0.04230751097202301, 0.0, 0.0, 0.0, 0.0, 0.0]}, "num_trajectories": 442226, "num_transitions": 7045476}, "nyu_franka_play_dataset_converted_externally_to_rlds": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.06424188613891602, 0.07027634978294373, 0.06129661202430725, 6.281067848205566, 0.1967729926109314, 0.26377415657043457, 1.0], "mean": [0.0010219913674518466, -0.00012002612493233755, 0.0003289413289166987, 0.0015034287935122848, -0.0021985324565321207, -0.0016632309416309, 0.7230083346366882], "min": [-0.05952230095863342, -0.07232445478439331, -0.06730806827545166, -6.278434753417969, -0.21479034423828125, -0.3627619743347168, 0.0], "p01": [-0.03199600800871849, -0.03286167234182358, -0.033688049763441086, -0.12080862373113632, -0.12175218015909195, -0.11370223760604858, 0.0], "p99": [0.031015200540423393, 0.03739088773727417, 0.036463744938373566, 0.11764093488454819, 0.1258920133113861, 0.09366150945425034, 1.0], "std": [0.01327415369451046, 0.013215922750532627, 0.012822109274566174, 0.27324536442756653, 0.05702253803610802, 0.03917280584573746, 0.4475320279598236]}, "num_trajectories": 456, "num_transitions": 44875}, "omnimimic_gnm_dataset": {"action": {"mask": [true, true], "max": [[10.04063606262207, 9.616271018981934], [10.025460243225098, 10.913275718688965], [9.976831436157227, 10.608161926269531], [9.830195426940918, 10.071197509765625], [9.676733016967773, 9.577154159545898], [9.494471549987793, 9.137883186340332], [9.505189895629883, 8.759404182434082], [9.769876480102539, 8.325521469116211], [9.945220947265625, 8.462895393371582], [10.027437210083008, 8.237380027770996], [10.040019035339355, 7.927189350128174], [9.999307632446289, 7.882480621337891], [9.924859046936035, 8.024871826171875], [10.019326210021973, 8.368189811706543], [9.877613067626953, 8.440105438232422], [9.306513786315918, 8.192524909973145], [8.926131248474121, 7.816405296325684], [8.933368682861328, 7.3515753746032715], [8.747367858886719, 7.082726955413818], [8.343022346496582, 7.5175299644470215], [8.055445671081543, 8.938166618347168], [8.084595680236816, 9.790791511535645], [8.19581127166748, 10.037925720214844], [8.323996543884277, 9.907886505126953], [8.088997840881348, 9.539944648742676], [7.330855369567871, 9.406754493713379], [6.941203594207764, 9.879721641540527], [6.756242275238037, 10.031426429748535], [6.5570244789123535, 10.031545639038086], [6.721526145935059, 9.80109691619873], [7.103391647338867, 9.575018882751465], [7.302797317504883, 9.734329223632812], [7.375810623168945, 9.846492767333984], [7.380764007568359, 9.899226188659668], [7.378841876983643, 9.880531311035156], [7.334626197814941, 9.87125015258789], [7.246583938598633, 9.836251258850098], [7.152384281158447, 9.743535041809082], [6.966617584228516, 9.447405815124512], [6.17810583114624, 9.31753921508789], [4.802942752838135, 9.545269966125488], [5.271764755249023, 9.722423553466797], [4.829601287841797, 9.824789047241211], [4.5858683586120605, 9.894800186157227], [4.543306827545166, 9.844291687011719], [4.760025978088379, 9.694314002990723], [6.268892288208008, 9.261051177978516], [7.933447360992432, 8.915355682373047], [9.130032539367676, 8.480704307556152], [9.521007537841797, 7.836717128753662], [9.174311637878418, 6.379136562347412], [9.151288986206055, 7.513707637786865], [10.064397811889648, 8.895575523376465], [10.689740180969238, 9.822613716125488], [10.91446590423584, 10.043546676635742], [10.673197746276855, 10.010782241821289], [10.039304733276367, 9.847620010375977], [9.777387619018555, 9.871333122253418], [10.847783088684082, 10.053329467773438], [10.640278816223145, 9.695094108581543], [9.886889457702637, 9.273459434509277], [9.451608657836914, 9.48863697052002], [10.277339935302734, 9.643360137939453], [10.810430526733398, 9.699681282043457], [10.914352416992188, 9.727193832397461], [10.828254699707031, 9.68575668334961], [10.619965553283691, 9.678448677062988], [10.740158081054688, 9.64224910736084], [10.88200855255127, 9.333720207214355], [10.909865379333496, 9.021982192993164], [10.884236335754395, 9.26081657409668], [10.75805950164795, 9.437880516052246], [10.508172988891602, 9.303016662597656], [10.264670372009277, 8.93529987335205], [10.174118995666504, 8.557365417480469], [10.57181453704834, 8.368375778198242], [10.88864803314209, 9.082667350769043], [10.866771697998047, 9.321856498718262], [10.84327220916748, 8.99880313873291], [10.79590892791748, 8.937424659729004], [10.597478866577148, 8.836201667785645], [9.631489753723145, 8.262746810913086], [9.728872299194336, 7.771418571472168], [10.050408363342285, 7.712123394012451], [10.026535034179688, 7.613933086395264], [9.974846839904785, 7.631654739379883], [9.501855850219727, 8.06674861907959], [6.968435287475586, 8.43349552154541], [5.975138187408447, 8.868706703186035], [6.639393329620361, 9.031000137329102], [7.092902660369873, 9.154200553894043], [7.341270446777344, 9.48193073272705], [7.369909286499023, 9.923832893371582], [7.236761093139648, 8.950642585754395], [6.931122779846191, 6.036386013031006], [6.4448723793029785, 5.186783790588379], [6.114978790283203, 4.50723934173584], [6.236913681030273, 3.853231906890869], [6.355235576629639, 3.053256034851074], [6.3785719871521, 4.284274578094482]], "mean": [[1.006650447845459, 0.02111281268298626], [0.9784367084503174, 0.02383464388549328], [0.9361762404441833, 0.02555396594107151], [0.8863998651504517, 0.026642216369509697], [0.8336644768714905, 0.027451902627944946], [0.7814456224441528, 0.028167370706796646], [0.731746256351471, 0.028888460248708725], [0.6856319308280945, 0.02970849722623825], [0.6436114311218262, 0.030543910339474678], [0.6060521602630615, 0.03152136504650116], [0.5727749466896057, 0.03265291452407837], [0.5430082082748413, 0.03386514261364937], [0.5159826874732971, 0.035042110830545425], [0.49102893471717834, 0.03614839166402817], [0.4678296744823456, 0.03714015334844589], [0.4460768401622772, 0.038042061030864716], [0.4255349040031433, 0.03880435600876808], [0.4060799777507782, 0.03949134051799774], [0.3876287043094635, 0.040131375193595886], [0.37008848786354065, 0.040703415870666504], [0.3533627390861511, 0.041280828416347504], [0.3374203145503998, 0.04188892990350723], [0.3222067654132843, 0.04248908907175064], [0.3077148199081421, 0.043033208698034286], [0.29384690523147583, 0.04357285052537918], [0.2806214392185211, 0.04403156787157059], [0.2680569291114807, 0.044409748166799545], [0.2560494542121887, 0.044730350375175476], [0.24461829662322998, 0.04497399181127548], [0.2337610274553299, 0.0450734868645668], [0.22344481945037842, 0.04504554718732834], [0.2136279195547104, 0.04489404335618019], [0.20428626239299774, 0.04470869526267052], [0.19534853100776672, 0.04444288834929466], [0.18684299290180206, 0.044091761112213135], [0.17876487970352173, 0.04375177621841431], [0.17106947302818298, 0.043397560715675354], [0.16369561851024628, 0.04299766570329666], [0.15663021802902222, 0.04252263531088829], [0.14986729621887207, 0.04198935255408287], [0.14337213337421417, 0.04137508571147919], [0.13720878958702087, 0.040707945823669434], [0.13132157921791077, 0.0400255024433136], [0.12567920982837677, 0.0393076129257679], [0.12031009793281555, 0.03860296308994293], [0.11523692309856415, 0.037906818091869354], [0.11038373410701752, 0.03718705102801323], [0.10574894398450851, 0.036471959203481674], [0.10132230818271637, 0.035747889429330826], [0.09711924940347672, 0.03501499816775322], [0.09314871579408646, 0.03432577848434448], [0.08941958099603653, 0.033650968223810196], [0.08583126217126846, 0.032922446727752686], [0.08241500705480576, 0.032162152230739594], [0.07913308590650558, 0.0314236544072628], [0.07603204995393753, 0.03064819984138012], [0.07310689985752106, 0.02983461134135723], [0.07026368379592896, 0.02900754287838936], [0.06756903231143951, 0.028272679075598717], [0.06498143076896667, 0.02758037857711315], [0.06251210719347, 0.026924077421426773], [0.06021439656615257, 0.026297926902770996], [0.058082230389118195, 0.025715233758091927], [0.05610940605401993, 0.025133125483989716], [0.054250381886959076, 0.02458084374666214], [0.05247167497873306, 0.024061579257249832], [0.0508074089884758, 0.023548437282443047], [0.049252450466156006, 0.023034600540995598], [0.04772746562957764, 0.02250036783516407], [0.046257589012384415, 0.02197878807783127], [0.04485642537474632, 0.021503465250134468], [0.04350950941443443, 0.021033069118857384], [0.04223733767867088, 0.02059360407292843], [0.04103732854127884, 0.0201443862169981], [0.039903491735458374, 0.019730491563677788], [0.03879852592945099, 0.01934262365102768], [0.037669822573661804, 0.018951142206788063], [0.03658673167228699, 0.018554668873548508], [0.035544224083423615, 0.018198611214756966], [0.034524623304605484, 0.017831888049840927], [0.03357917070388794, 0.017457369714975357], [0.03271720930933952, 0.0170982014387846], [0.031961869448423386, 0.016765428707003593], [0.031281013041734695, 0.016440996900200844], [0.030674442648887634, 0.016146911308169365], [0.030120784416794777, 0.015883535146713257], [0.029584158211946487, 0.01564367301762104], [0.029062815010547638, 0.015403839759528637], [0.028545772656798363, 0.01515283901244402], [0.02803557738661766, 0.01490924321115017], [0.027568122372031212, 0.014660588465631008], [0.027093244716525078, 0.014370201155543327], [0.026639770716428757, 0.014075732789933681], [0.026206808164715767, 0.013773839920759201], [0.02580174244940281, 0.01349083986133337], [0.025429967790842056, 0.013224533759057522], [0.02509704977273941, 0.012994644232094288], [0.024798430502414703, 0.012788897380232811], [0.024515300989151, 0.012543120421469212], [0.024188481271266937, 0.01230641733855009]], "min": [[-0.6939107775688171, -8.103500366210938], [-3.077446460723877, -9.130558013916016], [-5.095889091491699, -9.729989051818848], [-5.718104362487793, -9.999093055725098], [-5.982468605041504, -10.031413078308105], [-6.095625877380371, -9.745412826538086], [-6.511915683746338, -9.17488956451416], [-7.058178424835205, -8.390824317932129], [-7.625753402709961, -7.879674911499023], [-8.755743026733398, -7.672416687011719], [-9.786115646362305, -7.933297157287598], [-10.552811622619629, -8.83379077911377], [-10.89777946472168, -9.535879135131836], [-10.739481925964355, -9.644658088684082], [-10.056856155395508, -9.048726081848145], [-9.366582870483398, -8.281074523925781], [-9.317854881286621, -7.367784023284912], [-9.457338333129883, -6.547527313232422], [-9.579288482666016, -6.555565357208252], [-9.516718864440918, -6.34712553024292], [-9.384038925170898, -5.59437894821167], [-9.315620422363281, -6.206727027893066], [-9.87446117401123, -6.81552791595459], [-10.04995346069336, -6.8110833168029785], [-10.011055946350098, -6.964095115661621], [-10.03516674041748, -8.197760581970215], [-9.790793418884277, -8.98321533203125], [-9.597389221191406, -8.875617027282715], [-9.77313232421875, -8.574536323547363], [-9.932422637939453, -8.114073753356934], [-10.00333309173584, -8.123224258422852], [-9.828289985656738, -9.045012474060059], [-9.532654762268066, -9.7724609375], [-9.216633796691895, -10.333633422851562], [-8.77624797821045, -9.981002807617188], [-8.22775650024414, -9.603851318359375], [-7.441531181335449, -9.534062385559082], [-6.623698711395264, -10.502553939819336], [-7.549694061279297, -10.893755912780762], [-7.632549285888672, -10.760881423950195], [-6.721611499786377, -10.0770263671875], [-7.635820388793945, -9.557232856750488], [-8.798952102661133, -9.78811264038086], [-9.753305435180664, -9.904645919799805], [-10.055521011352539, -9.924241065979004], [-9.975961685180664, -9.822135925292969], [-9.83285140991211, -8.934941291809082], [-9.642156600952148, -7.496172904968262], [-9.281744956970215, -5.981194496154785], [-8.686768531799316, -5.336909770965576], [-8.598743438720703, -5.913052558898926], [-8.945857048034668, -5.948622226715088], [-8.719218254089355, -4.223752975463867], [-8.086219787597656, -2.967353343963623], [-7.359464645385742, -3.0502357482910156], [-7.348613262176514, -3.581376791000366], [-7.29996919631958, -3.0370826721191406], [-7.3629326820373535, -3.0689592361450195], [-7.030279636383057, -3.1251659393310547], [-6.288608074188232, -3.1842358112335205], [-5.70255708694458, -4.624025821685791], [-5.736237049102783, -5.458872318267822], [-5.370548725128174, -3.675389528274536], [-5.986166000366211, -3.1710050106048584], [-6.468411445617676, -3.1067054271698], [-6.853854179382324, -3.0781805515289307], [-7.250633239746094, -3.0996057987213135], [-7.287858963012695, -3.2195000648498535], [-7.543529033660889, -3.2326419353485107], [-7.705174446105957, -3.152613639831543], [-7.212009429931641, -3.179734230041504], [-6.738742351531982, -3.2019128799438477], [-6.107148170471191, -3.1995184421539307], [-5.488959312438965, -3.194075345993042], [-4.782869338989258, -3.2489912509918213], [-4.00698709487915, -3.2568955421447754], [-3.3363916873931885, -3.253889322280884], [-3.1878881454467773, -3.301168203353882], [-3.2817134857177734, -3.309187173843384], [-3.3609700202941895, -3.305018424987793], [-3.3145198822021484, -3.300414562225342], [-3.8316149711608887, -5.134840965270996], [-5.680273056030273, -7.355038642883301], [-6.675719261169434, -9.053448677062988], [-7.342881202697754, -10.321976661682129], [-7.684134006500244, -10.762816429138184], [-8.05644416809082, -10.904092788696289], [-8.227563858032227, -10.879143714904785], [-8.301741600036621, -10.835618019104004], [-8.372123718261719, -10.914592742919922], [-8.435273170471191, -10.913979530334473], [-8.455034255981445, -10.635700225830078], [-8.420129776000977, -9.799579620361328], [-8.38514232635498, -8.61677074432373], [-8.264161109924316, -7.129925727844238], [-9.662135124206543, -5.076941013336182], [-10.614656448364258, -4.136116981506348], [-10.914713859558105, -3.949852228164673], [-10.591997146606445, -3.75651478767395], [-10.03878402709961, -3.7167515754699707]], "p01": [[0.0, -0.24749034643173218], [0.0, -0.524893045425415], [0.0, -0.7546966671943665], [-0.0, -0.9477061033248901], [-0.07417848706245422, -1.090767741203308], [-0.318183034658432, -1.173102855682373], [-0.5591932535171509, -1.2208278179168701], [-0.7463324069976807, -1.2503132820129395], [-0.8699370622634888, -1.2666205167770386], [-0.9435842633247375, -1.2755755186080933], [-0.988107442855835, -1.2812987565994263], [-1.019838571548462, -1.2843092679977417], [-1.0461297035217285, -1.2866320610046387], [-1.0671178102493286, -1.287879467010498], [-1.0865254402160645, -1.288839340209961], [-1.1034374237060547, -1.2892628908157349], [-1.1188381910324097, -1.2909561395645142], [-1.1345857381820679, -1.2918660640716553], [-1.146926760673523, -1.2913975715637207], [-1.1576642990112305, -1.2909899950027466], [-1.1685552597045898, -1.2900632619857788], [-1.1782664060592651, -1.2892645597457886], [-1.1883594989776611, -1.2887381315231323], [-1.1966487169265747, -1.2890256643295288], [-1.205859661102295, -1.2887595891952515], [-1.213012933731079, -1.2871893644332886], [-1.2209924459457397, -1.2852798700332642], [-1.2291326522827148, -1.2835227251052856], [-1.2333877086639404, -1.2813667058944702], [-1.2377482652664185, -1.2801706790924072], [-1.2401132583618164, -1.2784544229507446], [-1.2430986166000366, -1.2787694931030273], [-1.2465592622756958, -1.2778445482254028], [-1.2499667406082153, -1.276495099067688], [-1.2547986507415771, -1.2750866413116455], [-1.2574094533920288, -1.271640658378601], [-1.259511947631836, -1.2710473537445068], [-1.2615225315093994, -1.270490050315857], [-1.26487135887146, -1.267826795578003], [-1.2670894861221313, -1.2660423517227173], [-1.2701560258865356, -1.2636172771453857], [-1.2725183963775635, -1.2615182399749756], [-1.276685357093811, -1.2597179412841797], [-1.2783653736114502, -1.2578810453414917], [-1.2804267406463623, -1.2569326162338257], [-1.2812769412994385, -1.2569732666015625], [-1.2830642461776733, -1.256361722946167], [-1.284055233001709, -1.255863904953003], [-1.2860572338104248, -1.2544927597045898], [-1.2879048585891724, -1.254027009010315], [-1.2883713245391846, -1.2525120973587036], [-1.2893825769424438, -1.2500243186950684], [-1.2877765893936157, -1.2485690116882324], [-1.2877589464187622, -1.2464739084243774], [-1.2874040603637695, -1.2445427179336548], [-1.2863826751708984, -1.2450659275054932], [-1.2870652675628662, -1.2438175678253174], [-1.2860099077224731, -1.2421057224273682], [-1.284967064857483, -1.2405158281326294], [-1.2847381830215454, -1.239623785018921], [-1.2841722965240479, -1.239736557006836], [-1.2826787233352661, -1.2374169826507568], [-1.282164454460144, -1.2375072240829468], [-1.2797991037368774, -1.2375017404556274], [-1.2761802673339844, -1.2360843420028687], [-1.2733720541000366, -1.2362010478973389], [-1.2716506719589233, -1.2346532344818115], [-1.268664002418518, -1.2352635860443115], [-1.2666016817092896, -1.2344043254852295], [-1.2639211416244507, -1.2333400249481201], [-1.2613298892974854, -1.232812762260437], [-1.257857084274292, -1.2317317724227905], [-1.255195140838623, -1.2299652099609375], [-1.253624439239502, -1.230538249015808], [-1.249995231628418, -1.228135347366333], [-1.2487006187438965, -1.2278202772140503], [-1.2462477684020996, -1.2255239486694336], [-1.2424837350845337, -1.22357177734375], [-1.2392834424972534, -1.2219539880752563], [-1.2386131286621094, -1.2194188833236694], [-1.2359414100646973, -1.2177006006240845], [-1.2349216938018799, -1.2147725820541382], [-1.2333416938781738, -1.2131669521331787], [-1.2307438850402832, -1.211358666419983], [-1.2283618450164795, -1.2091264724731445], [-1.2266324758529663, -1.2060978412628174], [-1.2244019508361816, -1.205608606338501], [-1.2199077606201172, -1.2045154571533203], [-1.2162727117538452, -1.2021510601043701], [-1.211881399154663, -1.199583649635315], [-1.2094998359680176, -1.1994067430496216], [-1.2060028314590454, -1.1971522569656372], [-1.2023733854293823, -1.1941782236099243], [-1.1992415189743042, -1.18984854221344], [-1.1970722675323486, -1.1861201524734497], [-1.1931111812591553, -1.183187484741211], [-1.1933045387268066, -1.1792351007461548], [-1.1927028894424438, -1.1752080917358398], [-1.190839409828186, -1.1710400581359863], [-1.1884249448776245, -1.1674422025680542]], "p99": [[1.8355309963226318, 0.26012128591537476], [1.7681407928466797, 0.6165795922279358], [1.7430938482284546, 0.8756921887397766], [1.7192518711090088, 1.054436206817627], [1.6984645128250122, 1.169884204864502], [1.6791237592697144, 1.2388261556625366], [1.6643894910812378, 1.2796016931533813], [1.651375651359558, 1.3040544986724854], [1.639101505279541, 1.3204537630081177], [1.6300697326660156, 1.330525279045105], [1.621229887008667, 1.336830496788025], [1.6114033460617065, 1.343373417854309], [1.6024969816207886, 1.3487062454223633], [1.5920088291168213, 1.3529179096221924], [1.5839636325836182, 1.3558837175369263], [1.5734643936157227, 1.3591285943984985], [1.564652681350708, 1.3614760637283325], [1.5525522232055664, 1.3648791313171387], [1.5421693325042725, 1.3660087585449219], [1.5315660238265991, 1.3678475618362427], [1.5218133926391602, 1.3699870109558105], [1.5116015672683716, 1.3706583976745605], [1.5000834465026855, 1.3722280263900757], [1.4898813962936401, 1.3734859228134155], [1.4813286066055298, 1.3747098445892334], [1.4735304117202759, 1.3752965927124023], [1.4662020206451416, 1.3748369216918945], [1.4590167999267578, 1.3757003545761108], [1.4517619609832764, 1.3759630918502808], [1.4437453746795654, 1.3747040033340454], [1.4373306035995483, 1.3738460540771484], [1.429940104484558, 1.3741717338562012], [1.4237377643585205, 1.3730640411376953], [1.417199730873108, 1.3727082014083862], [1.4123414754867554, 1.3704391717910767], [1.4081101417541504, 1.3675767183303833], [1.4032999277114868, 1.3665492534637451], [1.4001662731170654, 1.3648020029067993], [1.3975027799606323, 1.3614763021469116], [1.3947447538375854, 1.358310580253601], [1.391717553138733, 1.3562898635864258], [1.3907407522201538, 1.352612018585205], [1.3889529705047607, 1.349622130393982], [1.3878453969955444, 1.3452204465866089], [1.3863216638565063, 1.3422263860702515], [1.3852434158325195, 1.3402395248413086], [1.384523868560791, 1.3379850387573242], [1.3832931518554688, 1.3341715335845947], [1.3821592330932617, 1.3313218355178833], [1.3805315494537354, 1.3287832736968994], [1.379412055015564, 1.3256323337554932], [1.3786927461624146, 1.322540521621704], [1.377315640449524, 1.3192228078842163], [1.3763973712921143, 1.3158390522003174], [1.3757916688919067, 1.3127719163894653], [1.3741294145584106, 1.3104934692382812], [1.37373685836792, 1.3056421279907227], [1.3729372024536133, 1.302704930305481], [1.37215256690979, 1.2985506057739258], [1.3712992668151855, 1.2962052822113037], [1.370071291923523, 1.2946804761886597], [1.368520975112915, 1.2921758890151978], [1.3673310279846191, 1.2917667627334595], [1.366342544555664, 1.289583444595337], [1.364940881729126, 1.2866352796554565], [1.3637622594833374, 1.2860138416290283], [1.3619582653045654, 1.2832822799682617], [1.3601936101913452, 1.2823646068572998], [1.358276605606079, 1.2806577682495117], [1.3559051752090454, 1.2789525985717773], [1.3541173934936523, 1.2770252227783203], [1.3523337841033936, 1.2743014097213745], [1.3499705791473389, 1.2742867469787598], [1.3478848934173584, 1.2725577354431152], [1.345453143119812, 1.2717618942260742], [1.3443059921264648, 1.2716058492660522], [1.3431313037872314, 1.2693581581115723], [1.3413448333740234, 1.2685908079147339], [1.3394896984100342, 1.267492651939392], [1.338375449180603, 1.2656534910202026], [1.3369172811508179, 1.2628607749938965], [1.3349311351776123, 1.2610188722610474], [1.3336808681488037, 1.259743571281433], [1.3321641683578491, 1.2572466135025024], [1.3309091329574585, 1.2559009790420532], [1.328662395477295, 1.2551106214523315], [1.3269047737121582, 1.2533318996429443], [1.326127529144287, 1.2518936395645142], [1.324747085571289, 1.250633955001831], [1.3246934413909912, 1.249377727508545], [1.3233622312545776, 1.2467306852340698], [1.32211172580719, 1.244922161102295], [1.3206640481948853, 1.2417919635772705], [1.318696141242981, 1.2377508878707886], [1.3158538341522217, 1.2328336238861084], [1.313198447227478, 1.2302494049072266], [1.3121668100357056, 1.2269189357757568], [1.3111391067504883, 1.2252254486083984], [1.3101955652236938, 1.2220994234085083], [1.3082560300827026, 1.220582127571106]], "std": [[0.3489035367965698, 0.10035178810358047], [0.3605400621891022, 0.2227659523487091], [0.3802538514137268, 0.31954458355903625], [0.40403640270233154, 0.39753228425979614], [0.4300621449947357, 0.4567839801311493], [0.4578975439071655, 0.4986434280872345], [0.4834575653076172, 0.5275481343269348], [0.5055370926856995, 0.5475218892097473], [0.5232846736907959, 0.5615379810333252], [0.5377234816551208, 0.5714869499206543], [0.54648756980896, 0.5785712003707886], [0.5553510189056396, 0.5834953188896179], [0.5647396445274353, 0.586515486240387], [0.5704450607299805, 0.5880470275878906], [0.5756932497024536, 0.5884559154510498], [0.5795589685440063, 0.5880607962608337], [0.585606038570404, 0.5870421528816223], [0.5874449014663696, 0.5855322480201721], [0.5916438102722168, 0.5836290121078491], [0.5915870666503906, 0.5815136432647705], [0.5939545631408691, 0.5793113708496094], [0.5958048701286316, 0.5769986510276794], [0.5938498973846436, 0.5746949315071106], [0.5942453145980835, 0.5722682476043701], [0.5955273509025574, 0.5697658061981201], [0.5950152277946472, 0.5671201348304749], [0.5909332036972046, 0.5644723176956177], [0.589492917060852, 0.5616850256919861], [0.5894355177879333, 0.5587843656539917], [0.587162435054779, 0.5558532476425171], [0.5860207080841064, 0.5529643893241882], [0.5800561904907227, 0.5501478910446167], [0.5783095955848694, 0.5472608208656311], [0.5750842690467834, 0.5443147420883179], [0.5730985999107361, 0.5410908460617065], [0.5714154243469238, 0.537874162197113], [0.5696194767951965, 0.5346378087997437], [0.5672723054885864, 0.5314096808433533], [0.5651775598526001, 0.528198778629303], [0.5610605478286743, 0.5248702764511108], [0.5588285326957703, 0.5215032696723938], [0.556196391582489, 0.517982006072998], [0.5539917349815369, 0.5144002437591553], [0.5517551302909851, 0.5109178423881531], [0.5441322922706604, 0.5076144933700562], [0.5414157509803772, 0.5043575763702393], [0.5387939810752869, 0.5010983943939209], [0.5355189442634583, 0.49784401059150696], [0.532726526260376, 0.4946846067905426], [0.529792308807373, 0.4915282726287842], [0.526955783367157, 0.48828068375587463], [0.5241825580596924, 0.4850847125053406], [0.5184216499328613, 0.4818841516971588], [0.5154927372932434, 0.4787699282169342], [0.5125833749771118, 0.4756662845611572], [0.5094144344329834, 0.4725321829319], [0.5064632892608643, 0.4694724678993225], [0.5035122632980347, 0.4664677679538727], [0.5004262328147888, 0.46351394057273865], [0.49732521176338196, 0.46061599254608154], [0.4943123161792755, 0.45771411061286926], [0.4902850091457367, 0.4548601508140564], [0.4872635006904602, 0.4519742429256439], [0.48423898220062256, 0.4491637647151947], [0.481080025434494, 0.44650378823280334], [0.47791457176208496, 0.4438484013080597], [0.474894255399704, 0.44112980365753174], [0.47176340222358704, 0.43855103850364685], [0.4686450660228729, 0.4359951615333557], [0.46553128957748413, 0.4334821105003357], [0.46244314312934875, 0.43102097511291504], [0.45906710624694824, 0.4285767078399658], [0.45607665181159973, 0.42610079050064087], [0.4530871510505676, 0.4236487150192261], [0.4500748813152313, 0.4212752878665924], [0.44713959097862244, 0.41885700821876526], [0.4442680776119232, 0.4164027273654938], [0.4416709244251251, 0.4137166142463684], [0.4391261637210846, 0.4110127389431], [0.43654122948646545, 0.40838557481765747], [0.43391773104667664, 0.40582117438316345], [0.4313312768936157, 0.40328091382980347], [0.4287663698196411, 0.4008156657218933], [0.42610830068588257, 0.39846205711364746], [0.42342841625213623, 0.3960530757904053], [0.4209117293357849, 0.3936210870742798], [0.4182846248149872, 0.3913455605506897], [0.415635883808136, 0.389122873544693], [0.4130752682685852, 0.3868511915206909], [0.41059625148773193, 0.38454049825668335], [0.4081219434738159, 0.3822118937969208], [0.4057695269584656, 0.37986478209495544], [0.40351831912994385, 0.37743082642555237], [0.40138760209083557, 0.3748909533023834], [0.39918118715286255, 0.37234362959861755], [0.3970663845539093, 0.3698950707912445], [0.3950637876987457, 0.3674008250236511], [0.3930501341819763, 0.3648092746734619], [0.3910492956638336, 0.36228621006011963], [0.38904625177383423, 0.35982486605644226]]}, "num_trajectories": 28888, "num_transitions": 1334883}, "roboturk": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.39124172925949097, 0.4601028263568878, 0.4870833456516266, 1.816888689994812, 1.8240282535552979, 1.4824820756912231, 1.0], "mean": [0.0014448765432462096, -0.0015945174964144826, -0.0011753799626603723, 0.0023012321908026934, -0.000938242650590837, -0.00011485845607239753, 0.5746025443077087], "min": [-0.6546999216079712, -0.6365841031074524, -0.4217723608016968, -1.6695482730865479, -1.8023357391357422, -1.4630827903747559, 0.0], "p01": [-0.1342635452747345, -0.19996687769889832, -0.14829720556735992, -0.20720748603343964, -0.09676413983106613, -0.1807563453912735, 0.0], "p99": [0.14956976473331451, 0.1805950552225113, 0.18841814994812012, 0.21615414321422577, 0.09457383304834366, 0.1854330152273178, 1.0], "std": [0.04935372993350029, 0.06354570388793945, 0.0611649826169014, 0.09553473442792892, 0.08420121669769287, 0.06517895311117172, 0.4945080280303955]}, "num_trajectories": 1995, "num_transitions": 187507}, "stanford_hydra_dataset_converted_externally_to_rlds": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.02499854564666748, 0.02499903365969658, 0.024999922141432762, 0.24974457919597626, 0.24997030198574066, 0.24999946355819702, 1.0], "mean": [0.00077900814358145, 0.00013707710604649037, -0.0002548577613197267, 0.0012903371825814247, -0.004751719068735838, 0.002692894311621785, 0.48855218291282654], "min": [-0.024999044835567474, -0.024999700486660004, -0.02499929815530777, -0.24993225932121277, -0.2499666064977646, -0.2499932497739792, 0.0], "p01": [-0.019992006942629814, -0.02415412664413452, -0.02294175885617733, -0.11085530370473862, -0.12024571746587753, -0.1331477016210556, 0.0], "p99": [0.022886231541633606, 0.022358838468790054, 0.0241008959710598, 0.12370114773511887, 0.11323311924934387, 0.18474750220775604, 1.0], "std": [0.008022245019674301, 0.009131476283073425, 0.009574329480528831, 0.04122211039066315, 0.038430601358413696, 0.04606691002845764, 0.49978455901145935]}, "num_trajectories": 570, "num_transitions": 358234}, "taco_extra": {"action": {"mask": [true, true, true, true, true, true, false], "max": [1.5608147382736206, 2.172372579574585, 1.8152713775634766, 3.392831325531006, 4.247413158416748, 3.0427656173706055, 1.0], "mean": [-0.013198821805417538, 0.025906339287757874, -0.012259144335985184, 0.001321413554251194, 0.002324604196473956, 0.055225666612386703, 0.781635046005249], "min": [-1.6745959520339966, -1.8985087871551514, -1.725131869316101, -1.839556336402893, -3.38140869140625, -3.2894704341888428, 0.0], "p01": [-0.645061731338501, -0.7966006994247437, -0.6900959610939026, -0.5348972678184509, -0.7819103598594666, -0.699119508266449, 0.0], "p99": [0.6668227314949036, 1.0882542133331299, 0.792926013469696, 0.6061288118362427, 0.7010000348091125, 1.1651601791381836, 1.0], "std": [0.2259671539068222, 0.36724090576171875, 0.2521361708641052, 0.19997869431972504, 0.25952473282814026, 0.3282214403152466, 0.4132002294063568]}, "num_trajectories": 514, "num_transitions": 55192}, "taco_play": {"action": {"mask": [true, true, true, true, true, true, false], "max": [1.4915844202041626, 2.1842432022094727, 2.6836395263671875, 5.035226821899414, 2.665864944458008, 4.250768661499023, 1.0], "mean": [-0.0038459247443825006, 0.009671456180512905, 0.012780566699802876, -0.005403782706707716, -0.009606565348803997, -0.0024807178415358067, 0.4263913035392761], "min": [-4.242457866668701, -3.192805051803589, -1.3371467590332031, -4.202683448791504, -2.6722638607025146, -3.3467135429382324, 0.0], "p01": [-0.710614025592804, -1.0569446086883545, -0.5878450274467468, -0.7682853937149048, -0.7180147767066956, -1.5527938604354858, 0.0], "p99": [0.6482916474342346, 1.0051310062408447, 0.9480248689651489, 0.6926478147506714, 0.6351067423820496, 1.6280102729797363, 1.0], "std": [0.23254038393497467, 0.3629826605319977, 0.2869289517402649, 0.2617705166339874, 0.24388936161994934, 0.5216500163078308, 0.4946906864643097]}, "num_trajectories": 3603, "num_transitions": 237798}, "toto": {"action": {"mask": [true, true, true, true, true, true, false], "max": [0.6839867234230042, 0.4454185664653778, 0.7984078526496887, 2.120781660079956, 1.371164321899414, 1.4118704795837402, 0.0], "mean": [0.38542234897613525, 0.007769497577100992, 0.3632740080356598, -0.6652021408081055, 0.18903982639312744, 0.03298753872513771, 0.0], "min": [0.09922284632921219, -0.5180193781852722, 0.13791072368621826, -2.635117530822754, -1.0734480619430542, -1.9282547235488892, 0.0], "p01": [0.17567229270935059, -0.3077590763568878, 0.23538391292095184, -2.090850591659546, -0.6191593408584595, -0.7488683462142944, 0.0], "p99": [0.6136963367462158, 0.3370419442653656, 0.6681221723556519, 0.7422861456871033, 0.7955394983291626, 0.7404646277427673, 0.0], "std": [0.12211640179157257, 0.19378575682640076, 0.10178238898515701, 0.572525680065155, 0.2988455295562744, 0.32599157094955444, 0.0]}, "num_trajectories": 1003, "num_transitions": 325699}, "utaustin_mutex": {"action": {"mask": [true, true, true, true, true, true, false], "max": [1.0, 1.0, 1.0, 0.375, 0.375, 0.375, 1.0], "mean": [0.06176404282450676, -0.0050054555758833885, 0.10216785967350006, -0.03314128518104553, 0.013895018957555294, -0.011317629367113113, 0.5038976669311523], "min": [-1.0, -1.0, -1.0, -0.375, -0.375, -0.375, 0.0], "p01": [-0.4285714328289032, -0.9800000190734863, -0.5571428537368774, -0.375, -0.15642857551574707, -0.335357129573822, 0.0], "p99": [0.5914285778999329, 0.9714285731315613, 1.0, 0.3278571367263794, 0.207857146859169, 0.25607141852378845, 1.0], "std": [0.1875014752149582, 0.44684740900993347, 0.37928882241249084, 0.14097855985164642, 0.0645369216799736, 0.11765256524085999, 0.501045286655426]}, "num_trajectories": 1500, "num_transitions": 361883}, "viola": {"action": {"mask": [true, true, true, true, true, true, false], "max": [1.0, 1.0, 1.0, 0.375, 0.36321428418159485, 0.375, 1.0], "mean": [0.04761854559183121, -0.02920449525117874, 0.05586730316281319, -0.0026185207534581423, 0.006867343559861183, -0.01682135835289955, 0.7323777675628662], "min": [-1.0, -1.0, -1.0, -0.375, -0.375, -0.375, 0.0], "p01": [-0.9628571271896362, -1.0, -1.0, -0.26249998807907104, -0.21321429312229156, -0.3385714292526245, 0.0], "p99": [0.9114285707473755, 0.868571400642395, 1.0, 0.2817857265472412, 0.2239285707473755, 0.3557142913341522, 1.0], "std": [0.3915800154209137, 0.4076523184776306, 0.400779128074646, 0.10023969411849976, 0.08443202823400497, 0.10375082492828369, 0.44260546565055847]}, "num_trajectories": 150, "num_transitions": 76324}}
|
example_batch.msgpack
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f754caf846b6ad83f66a2076a2a1165e785dd0e67ab9f495bcf6743316d2fd5a
|
| 3 |
+
size 4556600
|