ruffy369 commited on
Commit
6c2163b
1 Parent(s): 7254642

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -2
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,
 
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,