Upload folder using huggingface_hub
Browse files- config.json +2 -2
- config.yaml +2 -22
- model.safetensors +2 -2
config.json
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
{
|
| 2 |
-
"chunk_size":
|
| 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":
|
| 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:
|
| 157 |
-
n_action_steps:
|
| 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:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d7847b0d632cb353031ca975be25f595f55aa0b4b930a122f3103ad67dff31e2
|
| 3 |
+
size 206463968
|