ar0s commited on
Commit
6a828c5
·
verified ·
1 Parent(s): 1952fcd

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. config.json +5 -5
  2. model.safetensors +1 -1
  3. train_config.json +7 -7
config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "type": "pi05",
3
- "n_obs_steps": 2,
4
  "input_features": {
5
  "observation.state": {
6
  "type": "STATE",
@@ -38,7 +38,7 @@
38
  "action_expert_variant": "gemma_300m",
39
  "dtype": "float32",
40
  "chunk_size": 50,
41
- "n_action_steps": 50,
42
  "max_state_dim": 32,
43
  "max_action_dim": 32,
44
  "num_inference_steps": 10,
@@ -60,11 +60,11 @@
60
  "STATE": "QUANTILES",
61
  "ACTION": "QUANTILES"
62
  },
63
- "gradient_checkpointing": false,
64
- "compile_model": false,
65
  "compile_mode": "max-autotune",
66
  "freeze_vision_encoder": false,
67
- "train_expert_only": true,
68
  "optimizer_lr": 2.5e-05,
69
  "optimizer_betas": [
70
  0.9,
 
1
  {
2
  "type": "pi05",
3
+ "n_obs_steps": 1,
4
  "input_features": {
5
  "observation.state": {
6
  "type": "STATE",
 
38
  "action_expert_variant": "gemma_300m",
39
  "dtype": "float32",
40
  "chunk_size": 50,
41
+ "n_action_steps": 10,
42
  "max_state_dim": 32,
43
  "max_action_dim": 32,
44
  "num_inference_steps": 10,
 
60
  "STATE": "QUANTILES",
61
  "ACTION": "QUANTILES"
62
  },
63
+ "gradient_checkpointing": true,
64
+ "compile_model": true,
65
  "compile_mode": "max-autotune",
66
  "freeze_vision_encoder": false,
67
+ "train_expert_only": false,
68
  "optimizer_lr": 2.5e-05,
69
  "optimizer_betas": [
70
  0.9,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0b00da5af49007cc4d0a368877494e6f2aebbf51cddbf89d03f161948ac613fc
3
  size 14467170760
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc31b3adb631d1e77d3a0421db65659554be8aa1b8f9b27e6ffe1d623895a548
3
  size 14467170760
train_config.json CHANGED
@@ -82,7 +82,7 @@
82
  "env": null,
83
  "policy": {
84
  "type": "pi05",
85
- "n_obs_steps": 2,
86
  "input_features": {
87
  "observation.state": {
88
  "type": "STATE",
@@ -120,7 +120,7 @@
120
  "action_expert_variant": "gemma_300m",
121
  "dtype": "float32",
122
  "chunk_size": 50,
123
- "n_action_steps": 50,
124
  "max_state_dim": 32,
125
  "max_action_dim": 32,
126
  "num_inference_steps": 10,
@@ -142,11 +142,11 @@
142
  "STATE": "QUANTILES",
143
  "ACTION": "QUANTILES"
144
  },
145
- "gradient_checkpointing": false,
146
- "compile_model": false,
147
  "compile_mode": "max-autotune",
148
  "freeze_vision_encoder": false,
149
- "train_expert_only": true,
150
  "optimizer_lr": 2.5e-05,
151
  "optimizer_betas": [
152
  0.9,
@@ -159,7 +159,7 @@
159
  "scheduler_decay_steps": 30000,
160
  "scheduler_decay_lr": 2.5e-06
161
  },
162
- "output_dir": "outputs/train/2026-02-18/14-10-25_pi05",
163
  "job_name": "pi05",
164
  "resume": false,
165
  "seed": 1000,
@@ -201,7 +201,7 @@
201
  "project": "lerobot",
202
  "entity": null,
203
  "notes": null,
204
- "run_id": "xd7hqt27",
205
  "mode": "online"
206
  },
207
  "peft": null,
 
82
  "env": null,
83
  "policy": {
84
  "type": "pi05",
85
+ "n_obs_steps": 1,
86
  "input_features": {
87
  "observation.state": {
88
  "type": "STATE",
 
120
  "action_expert_variant": "gemma_300m",
121
  "dtype": "float32",
122
  "chunk_size": 50,
123
+ "n_action_steps": 10,
124
  "max_state_dim": 32,
125
  "max_action_dim": 32,
126
  "num_inference_steps": 10,
 
142
  "STATE": "QUANTILES",
143
  "ACTION": "QUANTILES"
144
  },
145
+ "gradient_checkpointing": true,
146
+ "compile_model": true,
147
  "compile_mode": "max-autotune",
148
  "freeze_vision_encoder": false,
149
+ "train_expert_only": false,
150
  "optimizer_lr": 2.5e-05,
151
  "optimizer_betas": [
152
  0.9,
 
159
  "scheduler_decay_steps": 30000,
160
  "scheduler_decay_lr": 2.5e-06
161
  },
162
+ "output_dir": "outputs/train/2026-02-19/12-14-31_pi05",
163
  "job_name": "pi05",
164
  "resume": false,
165
  "seed": 1000,
 
201
  "project": "lerobot",
202
  "entity": null,
203
  "notes": null,
204
+ "run_id": "vsw3ekuq",
205
  "mode": "online"
206
  },
207
  "peft": null,