OliverHausdoerfer commited on
Commit
7dd59fd
·
verified ·
1 Parent(s): d975a66

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +3 -3
  2. config.json +2 -2
  3. model.safetensors +1 -1
  4. train_config.json +8 -8
README.md CHANGED
@@ -1,13 +1,13 @@
1
  ---
2
- datasets: pusht_merged_300_fine_relative_converted
3
  library_name: lerobot
4
  license: apache-2.0
5
  model_name: diffusion
6
  pipeline_tag: robotics
7
  tags:
8
- - robotics
9
- - lerobot
10
  - diffusion
 
 
11
  ---
12
 
13
  # Model Card for diffusion
 
1
  ---
2
+ datasets: pusht_merged_300_fine
3
  library_name: lerobot
4
  license: apache-2.0
5
  model_name: diffusion
6
  pipeline_tag: robotics
7
  tags:
 
 
8
  - diffusion
9
+ - lerobot
10
+ - robotics
11
  ---
12
 
13
  # Model Card for diffusion
config.json CHANGED
@@ -36,7 +36,7 @@
36
  "normalization_mapping": {
37
  "VISUAL": "MEAN_STD",
38
  "STATE": "MIN_MAX",
39
- "ACTION": "ASINH_MIN_MAX",
40
  "ENV": "MIN_MAX"
41
  },
42
  "drop_n_last_frames": 7,
@@ -70,7 +70,7 @@
70
  "clip_sample_range": 1.0,
71
  "num_inference_steps": null,
72
  "do_mask_loss_for_padding": false,
73
- "optimizer_lr": 2e-05,
74
  "optimizer_betas": [
75
  0.95,
76
  0.999
 
36
  "normalization_mapping": {
37
  "VISUAL": "MEAN_STD",
38
  "STATE": "MIN_MAX",
39
+ "ACTION": "MIN_MAX",
40
  "ENV": "MIN_MAX"
41
  },
42
  "drop_n_last_frames": 7,
 
70
  "clip_sample_range": 1.0,
71
  "num_inference_steps": null,
72
  "do_mask_loss_for_padding": false,
73
+ "optimizer_lr": 0.0002,
74
  "optimizer_betas": [
75
  0.95,
76
  0.999
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2bdbb1ccc6224e34e47a83dbcc5eb0c129647ba94abd7244431785618127e052
3
  size 995055496
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ee1307bf7930b9ffb9eac5c9720025a58df265163e09a95f6d9f8f000955c99
3
  size 995055496
train_config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "dataset": {
3
- "repo_id": "pusht_merged_300_fine_relative_converted",
4
- "root": "pusht_teleop_data/pusht_merged_300_fine_relative_converted",
5
  "episodes": null,
6
  "image_transforms": {
7
  "enable": false,
@@ -132,7 +132,7 @@
132
  "gaussian_blur": null,
133
  "CoverGreenT": null,
134
  "coarsity": "fine",
135
- "action_type": "relative"
136
  },
137
  "policy": {
138
  "type": "diffusion",
@@ -172,7 +172,7 @@
172
  "normalization_mapping": {
173
  "VISUAL": "MEAN_STD",
174
  "STATE": "MIN_MAX",
175
- "ACTION": "ASINH_MIN_MAX",
176
  "ENV": "MIN_MAX"
177
  },
178
  "drop_n_last_frames": 7,
@@ -206,7 +206,7 @@
206
  "clip_sample_range": 1.0,
207
  "num_inference_steps": null,
208
  "do_mask_loss_for_padding": false,
209
- "optimizer_lr": 2e-05,
210
  "optimizer_betas": [
211
  0.95,
212
  0.999
@@ -218,7 +218,7 @@
218
  "partial_green_t_cover_processor": false,
219
  "use_rnd": false
220
  },
221
- "output_dir": "outputs/train/2025-12-09/02-03-51_pusht_diffusion_fine",
222
  "job_name": "pusht_diffusion_fine",
223
  "resume": false,
224
  "seed": 1000,
@@ -232,7 +232,7 @@
232
  "use_policy_training_preset": true,
233
  "optimizer": {
234
  "type": "adam",
235
- "lr": 2e-05,
236
  "weight_decay": 1e-06,
237
  "grad_clip_norm": 10.0,
238
  "betas": [
@@ -259,7 +259,7 @@
259
  "project": "policies",
260
  "entity": null,
261
  "notes": null,
262
- "run_id": "yzpk2lg6",
263
  "mode": null
264
  },
265
  "checkpoint_path": null,
 
1
  {
2
  "dataset": {
3
+ "repo_id": "pusht_merged_300_fine",
4
+ "root": "pusht_teleop_data/pusht_merged_300_fine",
5
  "episodes": null,
6
  "image_transforms": {
7
  "enable": false,
 
132
  "gaussian_blur": null,
133
  "CoverGreenT": null,
134
  "coarsity": "fine",
135
+ "action_type": "absolute"
136
  },
137
  "policy": {
138
  "type": "diffusion",
 
172
  "normalization_mapping": {
173
  "VISUAL": "MEAN_STD",
174
  "STATE": "MIN_MAX",
175
+ "ACTION": "MIN_MAX",
176
  "ENV": "MIN_MAX"
177
  },
178
  "drop_n_last_frames": 7,
 
206
  "clip_sample_range": 1.0,
207
  "num_inference_steps": null,
208
  "do_mask_loss_for_padding": false,
209
+ "optimizer_lr": 0.0002,
210
  "optimizer_betas": [
211
  0.95,
212
  0.999
 
218
  "partial_green_t_cover_processor": false,
219
  "use_rnd": false
220
  },
221
+ "output_dir": "outputs/train/2025-12-10/09-43-55_pusht_diffusion_fine",
222
  "job_name": "pusht_diffusion_fine",
223
  "resume": false,
224
  "seed": 1000,
 
232
  "use_policy_training_preset": true,
233
  "optimizer": {
234
  "type": "adam",
235
+ "lr": 0.0002,
236
  "weight_decay": 1e-06,
237
  "grad_clip_norm": 10.0,
238
  "betas": [
 
259
  "project": "policies",
260
  "entity": null,
261
  "notes": null,
262
+ "run_id": "aux7jvgy",
263
  "mode": null
264
  },
265
  "checkpoint_path": null,