Hugging Face's logo

IliaLarchenko
/
dot_pusht_keypoints_best

Robotics
LeRobot
Safetensors
model_hub_mixin
pytorch_model_hub_mixin
dot
Model card Files
xet
Community
IliaLarchenko commited on
Commit
9586c39
·
verified ·
1 Parent(s): 9723a20

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +119 -45
config.json CHANGED
@@ -1,48 +1,122 @@
1
  {
2
- "alpha": 0.75,
3
- "crop_scale": 1.0,
4
- "dim_feedforward": 512,
5
- "dim_model": 128,
6
- "dropout": 0.1,
7
- "inference_horizon": 30,
8
- "input_normalization_modes": {
9
- "observation.environment_state": "min_max",
10
- "observation.state": "min_max"
11
- },
12
- "input_shapes": {
13
- "observation.environment_state": [
14
- 16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
  ],
16
- "observation.state": [
17
- 2
18
- ]
19
- },
20
- "lookback_aug": 5,
21
- "lookback_obs_steps": 10,
22
- "lora_rank": 20,
23
- "merge_lora": true,
24
- "n_decoder_layers": 8,
25
- "n_heads": 8,
26
- "n_obs_steps": 3,
27
- "noise_decay": 0.999995,
28
- "output_normalization_modes": {
29
- "action": "min_max"
30
- },
31
- "output_shapes": {
32
- "action": [
33
- 2
34
- ]
35
- },
36
- "pre_norm": true,
37
- "predict_every_n": 1,
38
- "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
39
- "rescale_shape": [
40
- 96,
41
- 96
42
- ],
43
- "return_every_n": 2,
44
- "state_noise": 0.01,
45
- "train_alpha": 0.9,
46
- "train_horizon": 30,
47
- "vision_backbone": "resnet18"
48
  }
 
1
  {
2
+ "type": "dot",
3
+ "n_obs_steps": 3,
4
+ "normalization_mapping": {
5
+ "VISUAL": "MEAN_STD",
6
+ "STATE": "MIN_MAX",
7
+ "ENV": "MIN_MAX",
8
+ "ACTION": "MIN_MAX"
9
+ },
10
+ "input_features": {
11
+ "observation.state": {
12
+ "type": "STATE",
13
+ "shape": [
14
+ 2
15
+ ]
16
+ },
17
+ "observation.environment_state": {
18
+ "type": "ENV",
19
+ "shape": [
20
+ 16
21
+ ]
22
+ }
23
+ },
24
+ "output_features": {
25
+ "action": {
26
+ "type": "ACTION",
27
+ "shape": [
28
+ 2
29
+ ]
30
+ }
31
+ },
32
+ "train_horizon": 30,
33
+ "inference_horizon": 30,
34
+ "lookback_obs_steps": 10,
35
+ "lookback_aug": 5,
36
+ "override_dataset_stats": false,
37
+ "new_dataset_stats": {
38
+ "action": {
39
+ "max": [
40
+ 512.0,
41
+ 512.0
42
+ ],
43
+ "min": [
44
+ 0.0,
45
+ 0.0
46
+ ]
47
+ },
48
+ "observation.environment_state": {
49
+ "max": [
50
+ 512.0,
51
+ 512.0,
52
+ 512.0,
53
+ 512.0,
54
+ 512.0,
55
+ 512.0,
56
+ 512.0,
57
+ 512.0,
58
+ 512.0,
59
+ 512.0,
60
+ 512.0,
61
+ 512.0,
62
+ 512.0,
63
+ 512.0,
64
+ 512.0,
65
+ 512.0
66
+ ],
67
+ "min": [
68
+ 0.0,
69
+ 0.0,
70
+ 0.0,
71
+ 0.0,
72
+ 0.0,
73
+ 0.0,
74
+ 0.0,
75
+ 0.0,
76
+ 0.0,
77
+ 0.0,
78
+ 0.0,
79
+ 0.0,
80
+ 0.0,
81
+ 0.0,
82
+ 0.0,
83
+ 0.0
84
+ ]
85
+ },
86
+ "observation.state": {
87
+ "max": [
88
+ 512.0,
89
+ 512.0
90
+ ],
91
+ "min": [
92
+ 0.0,
93
+ 0.0
94
+ ]
95
+ }
96
+ },
97
+ "vision_backbone": "resnet18",
98
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
99
+ "pre_norm": true,
100
+ "lora_rank": 20,
101
+ "merge_lora": false,
102
+ "dim_model": 128,
103
+ "n_heads": 8,
104
+ "dim_feedforward": 512,
105
+ "n_decoder_layers": 8,
106
+ "rescale_shape": [
107
+ 96,
108
+ 96
109
  ],
110
+ "crop_scale": 1.0,
111
+ "state_noise": 0.01,
112
+ "noise_decay": 0.999995,
113
+ "dropout": 0.1,
114
+ "alpha": 0.75,
115
+ "train_alpha": 0.9,
116
+ "predict_every_n": 1,
117
+ "return_every_n": 2,
118
+ "optimizer_lr": 0.0001,
119
+ "optimizer_min_lr": 0.0001,
120
+ "optimizer_lr_cycle_steps": 300000,
121
+ "optimizer_weight_decay": 1e-05
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
122
  }