Upload folder using huggingface_hub

config.json

@@ -1,5 +1,5 @@
 {
-    "type": "diffusion",
+    "type": "custom_diffusion",
     "n_obs_steps": 2,
     "input_features": {
         "observation.state": {
@@ -44,13 +44,13 @@
     "device": "cuda",
     "use_amp": false,
     "push_to_hub": true,
-    "repo_id": "ISdept/piper_arm",
+    "repo_id": null,
     "private": null,
     "tags": null,
     "license": null,
     "pretrained_path": null,
-    "horizon": 24,
-    "n_action_steps": 8,
+    "horizon": 16,
+    "n_action_steps": 16,
     "normalization_mapping": {
         "VISUAL": "MEAN_STD",
         "STATE": "MIN_MAX",
@@ -59,14 +59,14 @@
     "drop_n_last_frames": 7,
     "vision_backbone": "resnet18",
     "crop_shape": [
-        84,
-        84
+        400,
+        400
     ],
     "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": false,
+    "use_separate_rgb_encoder_per_camera": true,
     "down_dims": [
         512,
         1024,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:684d6306e3f4a3e264c436e583367660e0b107a0f1dbe6c4f7ca9ef2bd28d2c4
-size 1081485652
+oid sha256:cb9327629282103fce9fb65f1bc873342833b6fd0a8e47b62d4a4979de271dc4
+size 1171173468

policy_postprocessor_step_0_unnormalizer_processor.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f22a7e684a657bc8291024220a3fba3021eb231fc7ac018c8caa0f965323b28a
+oid sha256:3c2493497bce134e296270e7216ad52909bb9a5452b86246fbd144b3d91a4068
 size 10168

policy_preprocessor_step_3_normalizer_processor.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f22a7e684a657bc8291024220a3fba3021eb231fc7ac018c8caa0f965323b28a
+oid sha256:3c2493497bce134e296270e7216ad52909bb9a5452b86246fbd144b3d91a4068
 size 10168