Safetensors
Weiww99 commited on
Commit
21c2f19
·
verified ·
1 Parent(s): 3e0c7c4

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -29
config.json CHANGED
@@ -1,42 +1,15 @@
1
  {
2
  "type": "pi0",
3
  "n_obs_steps": 1,
4
- "normalization_mapping": {
5
- "VISUAL": "IDENTITY",
6
- "STATE": "MEAN_STD",
7
- "ACTION": "MEAN_STD"
8
- },
9
  "input_features": {},
10
  "output_features": {},
11
  "device": "cpu",
12
  "use_amp": false,
 
13
  "chunk_size": 50,
14
  "n_action_steps": 50,
15
  "max_state_dim": 75,
16
  "max_action_dim": 75,
17
- "resize_imgs_with_padding": [
18
- 224,
19
- 224
20
- ],
21
  "empty_cameras": 0,
22
- "adapt_to_pi_aloha": false,
23
- "use_delta_joint_actions_aloha": false,
24
- "tokenizer_max_length": 24,
25
- "proj_width": 768,
26
- "num_steps": 10,
27
- "use_cache": true,
28
- "attention_implementation": "flex",
29
- "freeze_vision_encoder": false,
30
- "train_expert_only": false,
31
- "train_state_proj": true,
32
- "optimizer_lr": 2.5e-05,
33
- "optimizer_betas": [
34
- 0.9,
35
- 0.95
36
- ],
37
- "optimizer_eps": 1e-08,
38
- "optimizer_weight_decay": 1e-10,
39
- "scheduler_warmup_steps": 1000,
40
- "scheduler_decay_steps": 30000,
41
- "scheduler_decay_lr": 2.5e-06
42
  }
 
1
  {
2
  "type": "pi0",
3
  "n_obs_steps": 1,
 
 
 
 
 
4
  "input_features": {},
5
  "output_features": {},
6
  "device": "cpu",
7
  "use_amp": false,
8
+ "dtype": "float32",
9
  "chunk_size": 50,
10
  "n_action_steps": 50,
11
  "max_state_dim": 75,
12
  "max_action_dim": 75,
 
 
 
 
13
  "empty_cameras": 0,
14
+ "tokenizer_max_length": 72
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
  }