Upload folder using huggingface_hub

components_epoch_1.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64f6564ea8727c60accaf115ca4217965941eb1c80e72d6d4d86fb35e8667258
+size 426867439

components_epoch_10.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c00f6fc3b69e7ae90cae6ce2041dafd00556e7fd17a711435302875264ff9a76
+size 426867571

components_epoch_2.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:408dc2d343a17593621f4200498bf46234c0d91d137baa6c7c00a385c9e9431d
+size 426867439

components_epoch_3.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0d81f0a2e571a19779ecdb9b1348a53fd1348f3c5a27202c28f4c3cd3ab81445
+size 426867439

components_epoch_4.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89f8707c748467cc39d5e2eaadc7cb850b6c934163f8bae25603315889d0e704
+size 426867439

components_epoch_5.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2b513d8620ae1ba3d842da18460bdf6e0c5b1ac6d18905aeb35a9ee71c7a18e5
+size 426867439

components_epoch_6.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ad5f8d177e2eff3df818cfe2e96826d8e7b0917113264faadfbfc4d2b202fb2b
+size 426867439

components_epoch_7.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b8e757b2842a69582473cc68b5970030f3747df735a86b18a3a55e7b9520a197
+size 426867439

components_epoch_8.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa3a969ecbcf7aa45d7df2a2a1c39533f76dea10fe80aef7fc792052994f9190
+size 426867439

components_epoch_9.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b238a6a2eed3a6864b11247a370e5dd1ceae1bfa05949d64ebe644d782b81ae7
+size 426867439

components_epoch_best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f91046a4508a73fa2f1d35774c8f8d97e3e51f07065e3b34f6f5e0015fcb2a0
+size 426867835

split_info.json

@@ -0,0 +1,1361 @@
+{
+  "seed": 42,
+  "val_fraction": 0.2,
+  "test_fraction": 0.2,
+  "train": [
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 38,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 19,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 26,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 77,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 106,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 81,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 38,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 39,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 35,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 26,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 22,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 11,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 9,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 33,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 112,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 24,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 0,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 31,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 28,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 12,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 37,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 47,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 17,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 82,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 12,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 34,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 44,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 14,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 113,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 15,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 29,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 2,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 64,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 36,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 3,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 98,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 41,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 16,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 18,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 13,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 28,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 48,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 8,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 6,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 10,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 31,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 32,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 38,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 48,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 109,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 32,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 4,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 34,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 63,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 35,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 1,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 36,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 47,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 101,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 7,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 40,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 44,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 25,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 54,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 40,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 15,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 102,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 80,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 8,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 31,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 14,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 83,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 27,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 22,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 43,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 69,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 11,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 118,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 20,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 62,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 16,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 43,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 22,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 35,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 18,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 45,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 4,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 30,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 68,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 1,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 53,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 90,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 94,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 41,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 93,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 42,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 116,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 14,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 7,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 25,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 59,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 43,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 74,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 27,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 58,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 49,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 49,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 17,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 10,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 49,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 27,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 92,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 38,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 40,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 75,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 21,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 20,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 96,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 31,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 17,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 117,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 16,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 11,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 19,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 67,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 34,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 88,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 41,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 91,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 24,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 97,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 23,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 26,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 86,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 25,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 55,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 39,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 24,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 87,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 108,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 8,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 40,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 36,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 24,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 1,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 2,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 37,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 44,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 30,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 45,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 56,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 107,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 19,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 9,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 46,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 13,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 50,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 23,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 13,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 119,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 111,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 47,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 15,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 16,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 46,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 44,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 52,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 71,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 114,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 5,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 20,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 12,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 57,
+      "label": 1.0
+    }
+  ],
+  "val": [
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 21,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 0,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 4,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 3,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 39,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 84,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 28,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 20,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 49,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 29,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 14,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 51,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 10,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 99,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 27,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 37,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 19,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 110,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 21,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 70,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 60,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 37,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 13,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 4,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 12,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 18,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 39,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 10,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 72,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 85,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 7,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 30,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 7,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 46,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 32,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 8,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 6,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 5,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 76,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 46,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 23,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 2,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 115,
+      "label": 1.0
+    }
+  ],
+  "test": [
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 18,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 48,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 66,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 5,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 11,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 61,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 1,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 30,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 9,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 104,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 41,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 36,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 5,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 29,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 9,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 33,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 6,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 95,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 42,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 3,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 28,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 33,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 100,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 47,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 29,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 42,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 6,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 2,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 103,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 43,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 73,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 33,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 48,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 78,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 105,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 45,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 34,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 26,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 32,
+      "label": 1.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 42,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 23,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 89,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 35,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 0,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 3,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+      "ep_idx": 15,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 45,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 25,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 21,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 65,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 17,
+      "label": 0.0
+    },
+    {
+      "source": "j-m-h/pick_place_clean_realsense_downscaled",
+      "ep_idx": 79,
+      "label": 1.0
+    },
+    {
+      "source": "fabiangrob/pick_place_task_fail_realsense_downscaled",
+      "ep_idx": 0,
+      "label": 0.0
+    },
+    {
+      "source": "fabiangrob/pick_place_extra_objects_realsense_downscaled",
+      "ep_idx": 22,
+      "label": 0.0
+    }
+  ]
+}

train_config.json

@@ -0,0 +1,32 @@
+{
+  "positive_repo_id": "j-m-h/pick_place_clean_realsense_downscaled",
+  "negative_repo_ids": [
+    "fabiangrob/pick_place_wrong_cube_realsense_downscaled",
+    "fabiangrob/pick_place_task_fail_realsense_downscaled",
+    "fabiangrob/pick_place_extra_objects_realsense_downscaled"
+  ],
+  "task_description": "pick up the orange cube and place it in the blue container",
+  "root": null,
+  "vlm_model_id": "ACIDE/FailSense-Calvin-2p-3b",
+  "output_dir": "checkpoints/fs_blocks_v2",
+  "num_epochs": 10,
+  "batch_size": 1,
+  "gradient_accumulation_steps": 4,
+  "lr": 0.0001,
+  "weight_decay": 0.1,
+  "val_fraction": 0.2,
+  "test_fraction": 0.2,
+  "seed": 42,
+  "device": "cuda",
+  "top_camera_key": "observation.images.top",
+  "wrist_camera_key": "observation.images.wrist",
+  "dry_run": false,
+  "video_backend": "pyav",
+  "wandb": false,
+  "wandb_project": "fs-block-training",
+  "wandb_run_name": null,
+  "n_train": 173,
+  "n_val": 43,
+  "n_test": 54,
+  "n_trainable_params": 106665993
+}

train_log.json

@@ -0,0 +1,66 @@
+[
+  {
+    "epoch": 1,
+    "train_loss": 0.499822,
+    "train_acc": 0.8035,
+    "val_acc": 0.9767
+  },
+  {
+    "epoch": 2,
+    "train_loss": 0.171729,
+    "train_acc": 0.9827,
+    "val_acc": 0.9767
+  },
+  {
+    "epoch": 3,
+    "train_loss": 0.07097,
+    "train_acc": 1.0,
+    "val_acc": 1.0
+  },
+  {
+    "epoch": 4,
+    "train_loss": 0.038944,
+    "train_acc": 1.0,
+    "val_acc": 1.0
+  },
+  {
+    "epoch": 5,
+    "train_loss": 0.035286,
+    "train_acc": 1.0,
+    "val_acc": 1.0
+  },
+  {
+    "epoch": 6,
+    "train_loss": 0.035589,
+    "train_acc": 1.0,
+    "val_acc": 1.0
+  },
+  {
+    "epoch": 7,
+    "train_loss": 0.031722,
+    "train_acc": 1.0,
+    "val_acc": 1.0
+  },
+  {
+    "epoch": 8,
+    "train_loss": 0.039997,
+    "train_acc": 1.0,
+    "val_acc": 1.0
+  },
+  {
+    "epoch": 9,
+    "train_loss": 0.025663,
+    "train_acc": 1.0,
+    "val_acc": 1.0
+  },
+  {
+    "epoch": 10,
+    "train_loss": 0.021916,
+    "train_acc": 1.0,
+    "val_acc": 1.0
+  },
+  {
+    "test_acc": 1.0,
+    "n_test": 54
+  }
+]