danielsanjosepro commited on
Commit
5199cd3
·
verified ·
1 Parent(s): e02bfbd

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +1 -1
  2. config.json +2 -2
  3. model.safetensors +1 -1
  4. train_config.json +5 -5
README.md CHANGED
@@ -5,9 +5,9 @@ license: apache-2.0
5
  model_name: temporal_consistent_flow
6
  pipeline_tag: robotics
7
  tags:
8
- - lerobot
9
  - robotics
10
  - temporal_consistent_flow
 
11
  ---
12
 
13
  # Model Card for temporal_consistent_flow
 
5
  model_name: temporal_consistent_flow
6
  pipeline_tag: robotics
7
  tags:
 
8
  - robotics
9
  - temporal_consistent_flow
10
+ - lerobot
11
  ---
12
 
13
  # Model Card for temporal_consistent_flow
config.json CHANGED
@@ -64,7 +64,7 @@
64
  "clip_sample_range": 1.0,
65
  "num_inference_steps": 100,
66
  "do_mask_loss_for_padding": false,
67
- "optimizer_lr": 0.0001,
68
  "optimizer_betas": [
69
  0.95,
70
  0.999
@@ -76,7 +76,7 @@
76
  "use_meanflow": true,
77
  "do_multi_step_sampling": false,
78
  "inference_timesteps": 100,
79
- "time_distribution": "uniform",
80
  "log_norm_mu": -0.4,
81
  "log_norm_sigma": 1.0,
82
  "use_autograd_functional_jvp": false,
 
64
  "clip_sample_range": 1.0,
65
  "num_inference_steps": 100,
66
  "do_mask_loss_for_padding": false,
67
+ "optimizer_lr": 0.000141,
68
  "optimizer_betas": [
69
  0.95,
70
  0.999
 
76
  "use_meanflow": true,
77
  "do_multi_step_sampling": false,
78
  "inference_timesteps": 100,
79
+ "time_distribution": "lognorm",
80
  "log_norm_mu": -0.4,
81
  "log_norm_sigma": 1.0,
82
  "use_autograd_functional_jvp": false,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:04c30e72acf6ebfc56612586ee2e4a4082d581e5370053f17384ef73ed6017cf
3
  size 214251472
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8423c5b334e25ba1f7ca61fabfc24bb15732ec4e67baaf1117223350eb3b4d5d
3
  size 214251472
train_config.json CHANGED
@@ -187,7 +187,7 @@
187
  "clip_sample_range": 1.0,
188
  "num_inference_steps": 100,
189
  "do_mask_loss_for_padding": false,
190
- "optimizer_lr": 0.0001,
191
  "optimizer_betas": [
192
  0.95,
193
  0.999
@@ -199,7 +199,7 @@
199
  "use_meanflow": true,
200
  "do_multi_step_sampling": false,
201
  "inference_timesteps": 100,
202
- "time_distribution": "uniform",
203
  "log_norm_mu": -0.4,
204
  "log_norm_sigma": 1.0,
205
  "use_autograd_functional_jvp": false,
@@ -213,7 +213,7 @@
213
  "action_batch_size": 2,
214
  "sampling_strategy": "deterministic"
215
  },
216
- "output_dir": "outputs/train/2025-11-01/20-27-54_pusht_temporal_consistent_flow",
217
  "job_name": "pusht_temporal_consistent_flow",
218
  "resume": false,
219
  "seed": 1000,
@@ -227,7 +227,7 @@
227
  "use_policy_training_preset": true,
228
  "optimizer": {
229
  "type": "adam",
230
- "lr": 0.0001,
231
  "weight_decay": 1e-06,
232
  "grad_clip_norm": 10.0,
233
  "betas": [
@@ -252,7 +252,7 @@
252
  "project": "policies",
253
  "entity": null,
254
  "notes": null,
255
- "run_id": "h0fj9dni",
256
  "mode": null
257
  },
258
  "checkpoint_path": null
 
187
  "clip_sample_range": 1.0,
188
  "num_inference_steps": 100,
189
  "do_mask_loss_for_padding": false,
190
+ "optimizer_lr": 0.000141,
191
  "optimizer_betas": [
192
  0.95,
193
  0.999
 
199
  "use_meanflow": true,
200
  "do_multi_step_sampling": false,
201
  "inference_timesteps": 100,
202
+ "time_distribution": "lognorm",
203
  "log_norm_mu": -0.4,
204
  "log_norm_sigma": 1.0,
205
  "use_autograd_functional_jvp": false,
 
213
  "action_batch_size": 2,
214
  "sampling_strategy": "deterministic"
215
  },
216
+ "output_dir": "outputs/train/2025-11-01/23-30-18_pusht_temporal_consistent_flow",
217
  "job_name": "pusht_temporal_consistent_flow",
218
  "resume": false,
219
  "seed": 1000,
 
227
  "use_policy_training_preset": true,
228
  "optimizer": {
229
  "type": "adam",
230
+ "lr": 0.000141,
231
  "weight_decay": 1e-06,
232
  "grad_clip_norm": 10.0,
233
  "betas": [
 
252
  "project": "policies",
253
  "entity": null,
254
  "notes": null,
255
+ "run_id": "il470squ",
256
  "mode": null
257
  },
258
  "checkpoint_path": null