gimarchetti commited on
Commit
ecdc4b6
·
verified ·
1 Parent(s): 8e99e3e

Upload policy weights, train config and readme

Browse files
Files changed (4) hide show
  1. README.md +2 -2
  2. config.json +1 -1
  3. model.safetensors +1 -1
  4. train_config.json +3 -3
README.md CHANGED
@@ -5,9 +5,9 @@ license: apache-2.0
5
  model_name: pi05
6
  pipeline_tag: robotics
7
  tags:
8
- - lerobot
9
- - pi05
10
  - robotics
 
 
11
  ---
12
 
13
  # Model Card for pi05
 
5
  model_name: pi05
6
  pipeline_tag: robotics
7
  tags:
 
 
8
  - robotics
9
+ - pi05
10
+ - lerobot
11
  ---
12
 
13
  # Model Card for pi05
config.json CHANGED
@@ -52,7 +52,7 @@
52
  "action_expert_variant": "gemma_300m",
53
  "dtype": "bfloat16",
54
  "chunk_size": 50,
55
- "n_action_steps": 10,
56
  "max_state_dim": 32,
57
  "max_action_dim": 32,
58
  "num_inference_steps": 10,
 
52
  "action_expert_variant": "gemma_300m",
53
  "dtype": "bfloat16",
54
  "chunk_size": 50,
55
+ "n_action_steps": 50,
56
  "max_state_dim": 32,
57
  "max_action_dim": 32,
58
  "num_inference_steps": 10,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1c533220796209198ed33e872b9063b19b26bfaca8918faf585de02421040637
3
  size 7473096344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:98484c5784e7842c09b5e00a3490490faac9d8c611b615290d4280ec49510cad
3
  size 7473096344
train_config.json CHANGED
@@ -134,7 +134,7 @@
134
  "action_expert_variant": "gemma_300m",
135
  "dtype": "bfloat16",
136
  "chunk_size": 50,
137
- "n_action_steps": 10,
138
  "max_state_dim": 32,
139
  "max_action_dim": 32,
140
  "num_inference_steps": 10,
@@ -174,7 +174,7 @@
174
  "scheduler_decay_lr": 2.5e-06
175
  },
176
  "output_dir": "ckpt/ur5-experiment-pi05",
177
- "job_name": "ur5-experiment-pi052026-03-06_02-51-59",
178
  "resume": false,
179
  "seed": 1000,
180
  "num_workers": 4,
@@ -215,7 +215,7 @@
215
  "project": "lerobot",
216
  "entity": null,
217
  "notes": null,
218
- "run_id": "vhevoz22",
219
  "mode": null
220
  },
221
  "peft": null,
 
134
  "action_expert_variant": "gemma_300m",
135
  "dtype": "bfloat16",
136
  "chunk_size": 50,
137
+ "n_action_steps": 50,
138
  "max_state_dim": 32,
139
  "max_action_dim": 32,
140
  "num_inference_steps": 10,
 
174
  "scheduler_decay_lr": 2.5e-06
175
  },
176
  "output_dir": "ckpt/ur5-experiment-pi05",
177
+ "job_name": "ur5-experiment-pi052026-03-07_03-00-58",
178
  "resume": false,
179
  "seed": 1000,
180
  "num_workers": 4,
 
215
  "project": "lerobot",
216
  "entity": null,
217
  "notes": null,
218
+ "run_id": "l31yailc",
219
  "mode": null
220
  },
221
  "peft": null,