Upload folder using huggingface_hub

config.json

@@ -48,7 +48,7 @@
     "private": null,
     "tags": null,
     "license": null,
-    "pretrained_path": "/content/outputs/train/output_diff4-0/checkpoints/014000/pretrained_model",
+    "pretrained_path": null,
     "horizon": 24,
     "n_action_steps": 8,
     "normalization_mapping": {
@@ -59,14 +59,14 @@
     "drop_n_last_frames": 7,
     "vision_backbone": "resnet18",
     "crop_shape": [
-        250,
-        400
+        84,
+        84
     ],
-    "crop_is_random": false,
+    "crop_is_random": true,
     "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
     "use_group_norm": false,
     "spatial_softmax_num_keypoints": 32,
-    "use_separate_rgb_encoder_per_camera": true,
+    "use_separate_rgb_encoder_per_camera": false,
     "down_dims": [
         512,
         1024,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:be15e9ec1fee40861f0ab64f8c5c323081d8693353a2a6f6cbb0993a588e85f7
-size 1353755228
+oid sha256:ef340c3efdbbab2feda7e9becff0df0746bb7215c5e59e76e0353340115d0639
+size 1264069004

train_config.json

@@ -130,7 +130,7 @@
         "private": null,
         "tags": null,
         "license": null,
-        "pretrained_path": "/content/outputs/train/output_diff4-0/checkpoints/014000/pretrained_model",
+        "pretrained_path": null,
         "horizon": 24,
         "n_action_steps": 8,
         "normalization_mapping": {
@@ -141,14 +141,14 @@
         "drop_n_last_frames": 7,
         "vision_backbone": "resnet18",
         "crop_shape": [
-            250,
-            400
+            84,
+            84
         ],
-        "crop_is_random": false,
+        "crop_is_random": true,
         "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
         "use_group_norm": false,
         "spatial_softmax_num_keypoints": 32,
-        "use_separate_rgb_encoder_per_camera": true,
+        "use_separate_rgb_encoder_per_camera": false,
         "down_dims": [
             512,
             1024,
@@ -178,7 +178,7 @@
         "scheduler_name": "cosine",
         "scheduler_warmup_steps": 500
     },
-    "output_dir": "outputs/train/output_diff4",
+    "output_dir": "outputs/train/output_diff5",
     "job_name": "pick_and_place",
     "resume": false,
     "seed": 1000,