ruffy369 commited on
Commit
e371cc0
1 Parent(s): 04653e0

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +4 -3
config.json CHANGED
@@ -18,15 +18,16 @@
18
  1,
19
  1
20
  ],
 
21
  "dropout": 0.0,
22
- "embed_dim_tokenizer": 512,
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
- "grad_acc_steps_tokenizer": 1,
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.42.0.dev0",
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,