Upload policy weights, train config and readme
Browse files- config.json +3 -1
- model.safetensors +2 -2
- train_config.json +5 -3
config.json
CHANGED
@@ -49,7 +49,7 @@
|
|
49 |
"pretrained_backbone_weights": null,
|
50 |
"use_group_norm": true,
|
51 |
"spatial_softmax_num_keypoints": 32,
|
52 |
-
"use_separate_rgb_encoder_per_camera":
|
53 |
"down_dims": [
|
54 |
512,
|
55 |
1024,
|
@@ -65,6 +65,8 @@
|
|
65 |
"sigma_min": 0.0001,
|
66 |
"clip_sample": true,
|
67 |
"clip_sample_range": 1.0,
|
|
|
|
|
68 |
"do_mask_loss_for_padding": false,
|
69 |
"optimizer_lr": 0.0001,
|
70 |
"optimizer_betas": [
|
|
|
49 |
"pretrained_backbone_weights": null,
|
50 |
"use_group_norm": true,
|
51 |
"spatial_softmax_num_keypoints": 32,
|
52 |
+
"use_separate_rgb_encoder_per_camera": true,
|
53 |
"down_dims": [
|
54 |
512,
|
55 |
1024,
|
|
|
65 |
"sigma_min": 0.0001,
|
66 |
"clip_sample": true,
|
67 |
"clip_sample_range": 1.0,
|
68 |
+
"fm_min_period": 0.004,
|
69 |
+
"fm_max_period": 4.0,
|
70 |
"do_mask_loss_for_padding": false,
|
71 |
"optimizer_lr": 0.0001,
|
72 |
"optimizer_betas": [
|
model.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:c69d7c6157a8c9ab50fddc9b57fec281ef2d90b8ad8e33e3bd3887e5255f2f84
|
3 |
+
size 1046090724
|
train_config.json
CHANGED
@@ -116,7 +116,7 @@
|
|
116 |
"pretrained_backbone_weights": null,
|
117 |
"use_group_norm": true,
|
118 |
"spatial_softmax_num_keypoints": 32,
|
119 |
-
"use_separate_rgb_encoder_per_camera":
|
120 |
"down_dims": [
|
121 |
512,
|
122 |
1024,
|
@@ -132,6 +132,8 @@
|
|
132 |
"sigma_min": 0.0001,
|
133 |
"clip_sample": true,
|
134 |
"clip_sample_range": 1.0,
|
|
|
|
|
135 |
"do_mask_loss_for_padding": false,
|
136 |
"optimizer_lr": 0.0001,
|
137 |
"optimizer_betas": [
|
@@ -143,7 +145,7 @@
|
|
143 |
"scheduler_name": "cosine",
|
144 |
"scheduler_warmup_steps": 500
|
145 |
},
|
146 |
-
"output_dir": "outputs/train/2025-09-
|
147 |
"job_name": "flow_matching",
|
148 |
"resume": false,
|
149 |
"seed": 1000,
|
@@ -182,7 +184,7 @@
|
|
182 |
"project": "lerobot",
|
183 |
"entity": null,
|
184 |
"notes": null,
|
185 |
-
"run_id": "
|
186 |
"mode": null
|
187 |
}
|
188 |
}
|
|
|
116 |
"pretrained_backbone_weights": null,
|
117 |
"use_group_norm": true,
|
118 |
"spatial_softmax_num_keypoints": 32,
|
119 |
+
"use_separate_rgb_encoder_per_camera": true,
|
120 |
"down_dims": [
|
121 |
512,
|
122 |
1024,
|
|
|
132 |
"sigma_min": 0.0001,
|
133 |
"clip_sample": true,
|
134 |
"clip_sample_range": 1.0,
|
135 |
+
"fm_min_period": 0.004,
|
136 |
+
"fm_max_period": 4.0,
|
137 |
"do_mask_loss_for_padding": false,
|
138 |
"optimizer_lr": 0.0001,
|
139 |
"optimizer_betas": [
|
|
|
145 |
"scheduler_name": "cosine",
|
146 |
"scheduler_warmup_steps": 500
|
147 |
},
|
148 |
+
"output_dir": "outputs/train/2025-09-07/07-38-37_flow_matching",
|
149 |
"job_name": "flow_matching",
|
150 |
"resume": false,
|
151 |
"seed": 1000,
|
|
|
184 |
"project": "lerobot",
|
185 |
"entity": null,
|
186 |
"notes": null,
|
187 |
+
"run_id": "g06zn1pi",
|
188 |
"mode": null
|
189 |
}
|
190 |
}
|