zijian2022 commited on
Commit
fefe7a1
·
verified ·
1 Parent(s): ef1fd2c

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. config.json +4 -4
  2. config.yaml +7 -7
  3. model.safetensors +2 -2
config.json CHANGED
@@ -5,17 +5,17 @@
5
  "dropout": 0.1,
6
  "feedforward_activation": "relu",
7
  "input_normalization_modes": {
8
- "observation.images.first": "mean_std",
9
- "observation.images.top": "mean_std",
10
  "observation.state": "mean_std"
11
  },
12
  "input_shapes": {
13
- "observation.images.first": [
14
  3,
15
  480,
16
  640
17
  ],
18
- "observation.images.top": [
19
  3,
20
  480,
21
  640
 
5
  "dropout": 0.1,
6
  "feedforward_activation": "relu",
7
  "input_normalization_modes": {
8
+ "observation.images.laptop": "mean_std",
9
+ "observation.images.phone": "mean_std",
10
  "observation.state": "mean_std"
11
  },
12
  "input_shapes": {
13
+ "observation.images.laptop": [
14
  3,
15
  480,
16
  640
17
  ],
18
+ "observation.images.phone": [
19
  3,
20
  480,
21
  640
config.yaml CHANGED
@@ -2,7 +2,7 @@ resume: false
2
  device: cuda
3
  use_amp: false
4
  seed: 1000
5
- dataset_repo_id: zijian2022/so100_test_llm2
6
  video_backend: pyav
7
  training:
8
  offline_steps: 80000
@@ -173,7 +173,7 @@ env:
173
  action_dim: 6
174
  fps: ${fps}
175
  override_dataset_stats:
176
- observation.images.top:
177
  mean:
178
  - - - 0.485
179
  - - - 0.456
@@ -182,7 +182,7 @@ override_dataset_stats:
182
  - - - 0.229
183
  - - - 0.224
184
  - - - 0.225
185
- observation.images.first:
186
  mean:
187
  - - - 0.485
188
  - - - 0.456
@@ -197,11 +197,11 @@ policy:
197
  chunk_size: 100
198
  n_action_steps: 100
199
  input_shapes:
200
- observation.images.top:
201
  - 3
202
  - 480
203
  - 640
204
- observation.images.first:
205
  - 3
206
  - 480
207
  - 640
@@ -211,8 +211,8 @@ policy:
211
  action:
212
  - ${env.action_dim}
213
  input_normalization_modes:
214
- observation.images.top: mean_std
215
- observation.images.first: mean_std
216
  observation.state: mean_std
217
  output_normalization_modes:
218
  action: mean_std
 
2
  device: cuda
3
  use_amp: false
4
  seed: 1000
5
+ dataset_repo_id: zijian2022/llm40
6
  video_backend: pyav
7
  training:
8
  offline_steps: 80000
 
173
  action_dim: 6
174
  fps: ${fps}
175
  override_dataset_stats:
176
+ observation.images.laptop:
177
  mean:
178
  - - - 0.485
179
  - - - 0.456
 
182
  - - - 0.229
183
  - - - 0.224
184
  - - - 0.225
185
+ observation.images.phone:
186
  mean:
187
  - - - 0.485
188
  - - - 0.456
 
197
  chunk_size: 100
198
  n_action_steps: 100
199
  input_shapes:
200
+ observation.images.laptop:
201
  - 3
202
  - 480
203
  - 640
204
+ observation.images.phone:
205
  - 3
206
  - 480
207
  - 640
 
211
  action:
212
  - ${env.action_dim}
213
  input_normalization_modes:
214
+ observation.images.laptop: mean_std
215
+ observation.images.phone: mean_std
216
  observation.state: mean_std
217
  output_normalization_modes:
218
  action: mean_std
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ec4c105bbcec5d909fdfa47374b199e28e406944e2f0fa2c959d452517bf00ec
3
- size 206701064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:112d2d541e33c60c1310b549ddde24a591733c3d4b760e245a7b51cde1498f5f
3
+ size 206701072