Upload folder using huggingface_hub

config.json

@@ -1,6 +1,6 @@
 {
     "type": "long_task_diffusion",
-    "n_obs_steps": 10,
+    "n_obs_steps": 4,
     "input_features": {
         "observation.state": {
             "type": "STATE",
@@ -49,21 +49,21 @@
     "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": "ResNet18_Weights.IMAGENET1K_V1",
+    "pretrained_backbone_weights": "ResNet50_Weights.IMAGENET1K_V2",
     "use_group_norm": false,
     "spatial_softmax_num_keypoints": 32,
     "use_separate_rgb_encoder_per_camera": true,
@@ -85,7 +85,7 @@
     "clip_sample": true,
     "clip_sample_range": 1.0,
     "num_inference_steps": null,
-    "do_mask_loss_for_padding": true,
+    "do_mask_loss_for_padding": false,
     "optimizer_lr": 0.0001,
     "optimizer_betas": [
         0.95,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:717485e59be72043fc8de6eb4167f524e562c3fea3d853771db2daf415bc0bd1
-size 1353755132
+oid sha256:67af634b0bec4b811c1c330fcb87f8924e0d0bcff91e2f9ef6ec77323c469364
+size 1365979908

policy_postprocessor_step_0_unnormalizer_processor.safetensors

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

policy_preprocessor_step_3_normalizer_processor.safetensors

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