IliaLarchenko commited on
Commit
3199689
·
verified ·
1 Parent(s): e278a94

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +121 -47
config.json CHANGED
@@ -1,50 +1,124 @@
1
  {
2
- "alpha": 0.98,
3
- "crop_scale": 0.8,
4
- "dim_feedforward": 512,
5
- "dim_model": 128,
6
- "dropout": 0.1,
7
- "inference_horizon": 100,
8
- "input_normalization_modes": {
9
- "observation.images.top": "mean_std",
10
- "observation.state": "min_max"
11
- },
12
- "input_shapes": {
13
- "observation.images.top": [
14
- 3,
15
- 480,
16
- 640
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  ],
18
- "observation.state": [
19
- 14
20
- ]
21
- },
22
- "lookback_aug": 5,
23
- "lookback_obs_steps": 30,
24
- "lora_rank": 20,
25
- "merge_lora": true,
26
- "n_decoder_layers": 8,
27
- "n_heads": 8,
28
- "n_obs_steps": 3,
29
- "noise_decay": 0.999995,
30
- "output_normalization_modes": {
31
- "action": "min_max"
32
- },
33
- "output_shapes": {
34
- "action": [
35
- 14
36
- ]
37
- },
38
- "pre_norm": true,
39
- "predict_every_n": 1,
40
- "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
41
- "rescale_shape": [
42
- 480,
43
- 640
44
- ],
45
- "return_every_n": 1,
46
- "state_noise": 0.01,
47
- "train_alpha": 0.99,
48
- "train_horizon": 150,
49
- "vision_backbone": "resnet18"
50
  }
 
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.images.top": {
12
+ "type": "VISUAL",
13
+ "shape": [
14
+ 3,
15
+ 480,
16
+ 640
17
+ ]
18
+ },
19
+ "observation.state": {
20
+ "type": "STATE",
21
+ "shape": [
22
+ 14
23
+ ]
24
+ }
25
+ },
26
+ "output_features": {
27
+ "action": {
28
+ "type": "ACTION",
29
+ "shape": [
30
+ 14
31
+ ]
32
+ }
33
+ },
34
+ "train_horizon": 150,
35
+ "inference_horizon": 100,
36
+ "lookback_obs_steps": 30,
37
+ "lookback_aug": 5,
38
+ "override_dataset_stats": false,
39
+ "new_dataset_stats": {
40
+ "action": {
41
+ "max": [
42
+ 512.0,
43
+ 512.0
44
+ ],
45
+ "min": [
46
+ 0.0,
47
+ 0.0
48
+ ]
49
+ },
50
+ "observation.environment_state": {
51
+ "max": [
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
+ 512.0,
67
+ 512.0
68
+ ],
69
+ "min": [
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
+ 0.0,
85
+ 0.0
86
+ ]
87
+ },
88
+ "observation.state": {
89
+ "max": [
90
+ 512.0,
91
+ 512.0
92
+ ],
93
+ "min": [
94
+ 0.0,
95
+ 0.0
96
+ ]
97
+ }
98
+ },
99
+ "vision_backbone": "resnet18",
100
+ "pretrained_backbone_weights": "ResNet18_Weights.IMAGENET1K_V1",
101
+ "pre_norm": true,
102
+ "lora_rank": 20,
103
+ "merge_lora": false,
104
+ "dim_model": 128,
105
+ "n_heads": 8,
106
+ "dim_feedforward": 512,
107
+ "n_decoder_layers": 8,
108
+ "rescale_shape": [
109
+ 480,
110
+ 640
111
  ],
112
+ "crop_scale": 1.0,
113
+ "state_noise": 0.01,
114
+ "noise_decay": 0.999995,
115
+ "dropout": 0.1,
116
+ "alpha": 0.98,
117
+ "train_alpha": 0.99,
118
+ "predict_every_n": 1,
119
+ "return_every_n": 1,
120
+ "optimizer_lr": 3e-05,
121
+ "optimizer_min_lr": 1e-05,
122
+ "optimizer_lr_cycle_steps": 100000,
123
+ "optimizer_weight_decay": 1e-05
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
124
  }