Upload folder using huggingface_hub
Browse files
config.json
CHANGED
|
@@ -28,8 +28,8 @@
|
|
| 28 |
"type": "VISUAL",
|
| 29 |
"shape": [
|
| 30 |
3,
|
| 31 |
-
|
| 32 |
-
|
| 33 |
]
|
| 34 |
}
|
| 35 |
},
|
|
@@ -43,18 +43,19 @@
|
|
| 43 |
},
|
| 44 |
"device": "cuda",
|
| 45 |
"use_amp": false,
|
|
|
|
| 46 |
"push_to_hub": true,
|
| 47 |
"repo_id": "ISdept/smolvla-piper",
|
| 48 |
"private": null,
|
| 49 |
"tags": null,
|
| 50 |
"license": null,
|
| 51 |
-
"pretrained_path": "
|
| 52 |
"chunk_size": 50,
|
| 53 |
"n_action_steps": 50,
|
| 54 |
"normalization_mapping": {
|
| 55 |
"VISUAL": "IDENTITY",
|
| 56 |
-
"STATE": "
|
| 57 |
-
"ACTION": "
|
| 58 |
},
|
| 59 |
"max_state_dim": 32,
|
| 60 |
"max_action_dim": 32,
|
|
|
|
| 28 |
"type": "VISUAL",
|
| 29 |
"shape": [
|
| 30 |
3,
|
| 31 |
+
400,
|
| 32 |
+
640
|
| 33 |
]
|
| 34 |
}
|
| 35 |
},
|
|
|
|
| 43 |
},
|
| 44 |
"device": "cuda",
|
| 45 |
"use_amp": false,
|
| 46 |
+
"use_peft": false,
|
| 47 |
"push_to_hub": true,
|
| 48 |
"repo_id": "ISdept/smolvla-piper",
|
| 49 |
"private": null,
|
| 50 |
"tags": null,
|
| 51 |
"license": null,
|
| 52 |
+
"pretrained_path": "lerobot/smolvla_base",
|
| 53 |
"chunk_size": 50,
|
| 54 |
"n_action_steps": 50,
|
| 55 |
"normalization_mapping": {
|
| 56 |
"VISUAL": "IDENTITY",
|
| 57 |
+
"STATE": "MEAN_STD",
|
| 58 |
+
"ACTION": "MEAN_STD"
|
| 59 |
},
|
| 60 |
"max_state_dim": 32,
|
| 61 |
"max_action_dim": 32,
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 906712520
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0b9045f761edb6048e6b7faca5941bbc77301942e07eea26fd4921e2ad115eb4
|
| 3 |
size 906712520
|
policy_postprocessor.json
CHANGED
|
@@ -15,8 +15,8 @@
|
|
| 15 |
},
|
| 16 |
"norm_map": {
|
| 17 |
"VISUAL": "IDENTITY",
|
| 18 |
-
"STATE": "
|
| 19 |
-
"ACTION": "
|
| 20 |
}
|
| 21 |
},
|
| 22 |
"state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
|
|
|
|
| 15 |
},
|
| 16 |
"norm_map": {
|
| 17 |
"VISUAL": "IDENTITY",
|
| 18 |
+
"STATE": "MEAN_STD",
|
| 19 |
+
"ACTION": "MEAN_STD"
|
| 20 |
}
|
| 21 |
},
|
| 22 |
"state_file": "policy_postprocessor_step_0_unnormalizer_processor.safetensors"
|
policy_postprocessor_step_0_unnormalizer_processor.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:597bd1f1e20b9057f3a28b3772fd5b2591a3fba01f55466f463b2e0ed2908f10
|
| 3 |
+
size 4400
|
policy_preprocessor.json
CHANGED
|
@@ -5,8 +5,9 @@
|
|
| 5 |
"registry_name": "rename_observations_processor",
|
| 6 |
"config": {
|
| 7 |
"rename_map": {
|
| 8 |
-
"observation.images.
|
| 9 |
-
"observation.images.gripper": "observation.images.camera2"
|
|
|
|
| 10 |
}
|
| 11 |
}
|
| 12 |
},
|
|
@@ -67,8 +68,8 @@
|
|
| 67 |
"type": "VISUAL",
|
| 68 |
"shape": [
|
| 69 |
3,
|
| 70 |
-
|
| 71 |
-
|
| 72 |
]
|
| 73 |
},
|
| 74 |
"action": {
|
|
@@ -80,8 +81,8 @@
|
|
| 80 |
},
|
| 81 |
"norm_map": {
|
| 82 |
"VISUAL": "IDENTITY",
|
| 83 |
-
"STATE": "
|
| 84 |
-
"ACTION": "
|
| 85 |
}
|
| 86 |
},
|
| 87 |
"state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors"
|
|
|
|
| 5 |
"registry_name": "rename_observations_processor",
|
| 6 |
"config": {
|
| 7 |
"rename_map": {
|
| 8 |
+
"observation.images.front": "observation.images.camera1",
|
| 9 |
+
"observation.images.gripper": "observation.images.camera2",
|
| 10 |
+
"observation.images.right": "observation.images.camera3"
|
| 11 |
}
|
| 12 |
}
|
| 13 |
},
|
|
|
|
| 68 |
"type": "VISUAL",
|
| 69 |
"shape": [
|
| 70 |
3,
|
| 71 |
+
400,
|
| 72 |
+
640
|
| 73 |
]
|
| 74 |
},
|
| 75 |
"action": {
|
|
|
|
| 81 |
},
|
| 82 |
"norm_map": {
|
| 83 |
"VISUAL": "IDENTITY",
|
| 84 |
+
"STATE": "MEAN_STD",
|
| 85 |
+
"ACTION": "MEAN_STD"
|
| 86 |
}
|
| 87 |
},
|
| 88 |
"state_file": "policy_preprocessor_step_5_normalizer_processor.safetensors"
|
policy_preprocessor_step_5_normalizer_processor.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:597bd1f1e20b9057f3a28b3772fd5b2591a3fba01f55466f463b2e0ed2908f10
|
| 3 |
+
size 4400
|
train_config.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"dataset": {
|
| 3 |
-
"repo_id": "ISdept/piper-pick-place
|
| 4 |
"root": null,
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
|
@@ -110,8 +110,8 @@
|
|
| 110 |
"type": "VISUAL",
|
| 111 |
"shape": [
|
| 112 |
3,
|
| 113 |
-
|
| 114 |
-
|
| 115 |
]
|
| 116 |
}
|
| 117 |
},
|
|
@@ -131,13 +131,13 @@
|
|
| 131 |
"private": null,
|
| 132 |
"tags": null,
|
| 133 |
"license": null,
|
| 134 |
-
"pretrained_path": "
|
| 135 |
"chunk_size": 50,
|
| 136 |
"n_action_steps": 50,
|
| 137 |
"normalization_mapping": {
|
| 138 |
"VISUAL": "IDENTITY",
|
| 139 |
-
"STATE": "
|
| 140 |
-
"ACTION": "
|
| 141 |
},
|
| 142 |
"max_state_dim": 32,
|
| 143 |
"max_action_dim": 32,
|
|
@@ -179,7 +179,7 @@
|
|
| 179 |
"max_period": 4.0,
|
| 180 |
"rtc_config": null
|
| 181 |
},
|
| 182 |
-
"output_dir": "outputs/train/
|
| 183 |
"job_name": "smolvla",
|
| 184 |
"resume": false,
|
| 185 |
"seed": 1000,
|
|
@@ -231,8 +231,9 @@
|
|
| 231 |
"rabc_epsilon": 1e-06,
|
| 232 |
"rabc_head_mode": "sparse",
|
| 233 |
"rename_map": {
|
| 234 |
-
"observation.images.
|
| 235 |
-
"observation.images.gripper": "observation.images.camera2"
|
|
|
|
| 236 |
},
|
| 237 |
"checkpoint_path": null
|
| 238 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"dataset": {
|
| 3 |
+
"repo_id": "ISdept/20260204-piper-pick-place",
|
| 4 |
"root": null,
|
| 5 |
"episodes": null,
|
| 6 |
"image_transforms": {
|
|
|
|
| 110 |
"type": "VISUAL",
|
| 111 |
"shape": [
|
| 112 |
3,
|
| 113 |
+
400,
|
| 114 |
+
640
|
| 115 |
]
|
| 116 |
}
|
| 117 |
},
|
|
|
|
| 131 |
"private": null,
|
| 132 |
"tags": null,
|
| 133 |
"license": null,
|
| 134 |
+
"pretrained_path": "lerobot/smolvla_base",
|
| 135 |
"chunk_size": 50,
|
| 136 |
"n_action_steps": 50,
|
| 137 |
"normalization_mapping": {
|
| 138 |
"VISUAL": "IDENTITY",
|
| 139 |
+
"STATE": "MEAN_STD",
|
| 140 |
+
"ACTION": "MEAN_STD"
|
| 141 |
},
|
| 142 |
"max_state_dim": 32,
|
| 143 |
"max_action_dim": 32,
|
|
|
|
| 179 |
"max_period": 4.0,
|
| 180 |
"rtc_config": null
|
| 181 |
},
|
| 182 |
+
"output_dir": "outputs/train/output_smolvla_7dof_5",
|
| 183 |
"job_name": "smolvla",
|
| 184 |
"resume": false,
|
| 185 |
"seed": 1000,
|
|
|
|
| 231 |
"rabc_epsilon": 1e-06,
|
| 232 |
"rabc_head_mode": "sparse",
|
| 233 |
"rename_map": {
|
| 234 |
+
"observation.images.front": "observation.images.camera1",
|
| 235 |
+
"observation.images.gripper": "observation.images.camera2",
|
| 236 |
+
"observation.images.right": "observation.images.camera3"
|
| 237 |
},
|
| 238 |
"checkpoint_path": null
|
| 239 |
}
|