andabi commited on
Commit
0cf0f47
·
verified ·
1 Parent(s): f3b147e

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. config.json +2 -2
  2. config.yaml +2 -22
  3. model.safetensors +2 -2
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "chunk_size": 50,
3
  "dim_feedforward": 3200,
4
  "dim_model": 512,
5
  "dropout": 0.1,
@@ -32,7 +32,7 @@
32
  },
33
  "kl_weight": 10.0,
34
  "latent_dim": 32,
35
- "n_action_steps": 50,
36
  "n_decoder_layers": 1,
37
  "n_encoder_layers": 4,
38
  "n_heads": 8,
 
1
  {
2
+ "chunk_size": 30,
3
  "dim_feedforward": 3200,
4
  "dim_model": 512,
5
  "dropout": 0.1,
 
32
  },
33
  "kl_weight": 10.0,
34
  "latent_dim": 32,
35
+ "n_action_steps": 30,
36
  "n_decoder_layers": 1,
37
  "n_encoder_layers": 4,
38
  "n_heads": 8,
config.yaml CHANGED
@@ -86,26 +86,6 @@ training:
86
  - 1.8
87
  - 1.8666666666666667
88
  - 1.9333333333333333
89
- - 2.0
90
- - 2.066666666666667
91
- - 2.1333333333333333
92
- - 2.2
93
- - 2.2666666666666666
94
- - 2.3333333333333335
95
- - 2.4
96
- - 2.466666666666667
97
- - 2.533333333333333
98
- - 2.6
99
- - 2.6666666666666665
100
- - 2.7333333333333334
101
- - 2.8
102
- - 2.8666666666666667
103
- - 2.933333333333333
104
- - 3.0
105
- - 3.066666666666667
106
- - 3.1333333333333333
107
- - 3.2
108
- - 3.2666666666666666
109
  eval:
110
  n_episodes: 50
111
  batch_size: 50
@@ -153,8 +133,8 @@ override_dataset_stats:
153
  policy:
154
  name: act
155
  n_obs_steps: 1
156
- chunk_size: 50
157
- n_action_steps: 50
158
  input_shapes:
159
  observation.images.bird:
160
  - 3
 
86
  - 1.8
87
  - 1.8666666666666667
88
  - 1.9333333333333333
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
89
  eval:
90
  n_episodes: 50
91
  batch_size: 50
 
133
  policy:
134
  name: act
135
  n_obs_steps: 1
136
+ chunk_size: 30
137
+ n_action_steps: 30
138
  input_shapes:
139
  observation.images.bird:
140
  - 3
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6f8e0d4dcb9a620d299e7ed027b8e2967e38801504db6ed3df667a8292fd909d
3
- size 206545888
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7847b0d632cb353031ca975be25f595f55aa0b4b930a122f3103ad67dff31e2
3
+ size 206463968