Upload folder using huggingface_hub

config.json

@@ -49,22 +49,22 @@
     "tags": null,
     "license": null,
     "pretrained_path": null,
-    "horizon": 16,
-    "n_action_steps": 8,
+    "horizon": 8,
+    "n_action_steps": 4,
     "normalization_mapping": {
         "VISUAL": "MEAN_STD",
         "STATE": "MIN_MAX",
         "ACTION": "MIN_MAX"
     },
     "drop_n_last_frames": 7,
-    "vision_backbone": "resnet18",
+    "vision_backbone": "resnet50",
     "crop_shape": [
         320,
         320
     ],
     "crop_is_random": true,
-    "pretrained_backbone_weights": null,
-    "use_group_norm": true,
+    "pretrained_backbone_weights": "ResNet50_Weights.IMAGENET1K_V2",
+    "use_group_norm": false,
     "spatial_softmax_num_keypoints": 32,
     "use_separate_rgb_encoder_per_camera": true,
     "down_dims": [

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ca77a9c6f1d50e7b09809a0449ddad9c4dab0882ee014d1a50de739f2c662cfb
-size 1216680540
+oid sha256:be1f374791e08220d6a1fd840c91ea520755e323534a57dd012224eff4b6c7d3
+size 1365979908