kelvinzhaozg commited on
Commit
15828e3
·
verified ·
1 Parent(s): 4f7c07a

Upload policy weights, train config and readme

Browse files
Files changed (3) hide show
  1. config.json +3 -1
  2. model.safetensors +2 -2
  3. 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": false,
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:0e5374df80417d66de57cb9dbb13dffbb810faacf28b6756af9607586448df7d
3
- size 1046090228
 
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": false,
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-06/15-09-30_flow_matching",
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": "zxhpto0x",
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
  }