Update config
Browse files- config.json +2 -4
config.json
CHANGED
|
@@ -24,7 +24,6 @@
|
|
| 24 |
"embed_pdrop": 0.1,
|
| 25 |
"entropy_weight": 0.001,
|
| 26 |
"eos_token_id": 50256,
|
| 27 |
-
"epochs": 600,
|
| 28 |
"gamma": 0.995,
|
| 29 |
"grad_acc_steps_actor_critic": 1,
|
| 30 |
"grad_acc_steps_tokenizer": 1,
|
|
@@ -44,11 +43,10 @@
|
|
| 44 |
"resid_pdrop": 0.1,
|
| 45 |
"resolution": 64,
|
| 46 |
"sequence_length": 20,
|
| 47 |
-
"should_sample_collect_test": true,
|
| 48 |
"tokens_per_block": 17,
|
| 49 |
-
"transformers_version": "4.
|
|
|
|
| 50 |
"use_original_obs_actor_critic": false,
|
| 51 |
"vocab_size": 512,
|
| 52 |
-
"weight_decay": 0.01,
|
| 53 |
"z_channels": 512
|
| 54 |
}
|
|
|
|
| 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 |
"grad_acc_steps_tokenizer": 1,
|
|
|
|
| 43 |
"resid_pdrop": 0.1,
|
| 44 |
"resolution": 64,
|
| 45 |
"sequence_length": 20,
|
|
|
|
| 46 |
"tokens_per_block": 17,
|
| 47 |
+
"transformers_version": "4.42.0.dev0",
|
| 48 |
+
"use_cache": true,
|
| 49 |
"use_original_obs_actor_critic": false,
|
| 50 |
"vocab_size": 512,
|
|
|
|
| 51 |
"z_channels": 512
|
| 52 |
}
|