Update config.json
Browse files- config.json +4 -3
config.json
CHANGED
|
@@ -18,15 +18,16 @@
|
|
| 18 |
1,
|
| 19 |
1
|
| 20 |
],
|
|
|
|
| 21 |
"dropout": 0.0,
|
| 22 |
-
"
|
| 23 |
"embed_dim_world_model": 256,
|
| 24 |
"embed_pdrop": 0.1,
|
| 25 |
"entropy_weight": 0.001,
|
| 26 |
"eos_token_id": 50256,
|
| 27 |
"gamma": 0.995,
|
| 28 |
"grad_acc_steps_actor_critic": 1,
|
| 29 |
-
"
|
| 30 |
"grad_acc_steps_world_model": 1,
|
| 31 |
"imagine_horizon_train_actor_critic": 20,
|
| 32 |
"in_channels": 3,
|
|
@@ -45,7 +46,7 @@
|
|
| 45 |
"resolution": 64,
|
| 46 |
"sequence_length": 20,
|
| 47 |
"tokens_per_block": 17,
|
| 48 |
-
"transformers_version": "4.
|
| 49 |
"use_cache": true,
|
| 50 |
"use_original_obs_actor_critic": false,
|
| 51 |
"vocab_size": 512,
|
|
|
|
| 18 |
1,
|
| 19 |
1
|
| 20 |
],
|
| 21 |
+
"device": "cuda:0",
|
| 22 |
"dropout": 0.0,
|
| 23 |
+
"embed_dim_discrete_autoencoder": 512,
|
| 24 |
"embed_dim_world_model": 256,
|
| 25 |
"embed_pdrop": 0.1,
|
| 26 |
"entropy_weight": 0.001,
|
| 27 |
"eos_token_id": 50256,
|
| 28 |
"gamma": 0.995,
|
| 29 |
"grad_acc_steps_actor_critic": 1,
|
| 30 |
+
"grad_acc_steps_discrete_autoencoder": 1,
|
| 31 |
"grad_acc_steps_world_model": 1,
|
| 32 |
"imagine_horizon_train_actor_critic": 20,
|
| 33 |
"in_channels": 3,
|
|
|
|
| 46 |
"resolution": 64,
|
| 47 |
"sequence_length": 20,
|
| 48 |
"tokens_per_block": 17,
|
| 49 |
+
"transformers_version": "4.44.0.dev0",
|
| 50 |
"use_cache": true,
|
| 51 |
"use_original_obs_actor_critic": false,
|
| 52 |
"vocab_size": 512,
|