ruffy369
/

iris-krull

Reinforcement Learning

Model card Files Files and versions

ruffy369 commited on Jul 25, 2024

Commit

6c2163b

·

verified ·

1 Parent(s): 7254642

Update config.json

Files changed (1) hide show

config.json +3 -2

config.json CHANGED Viewed

@@ -18,15 +18,16 @@
     1,
     1
   ],
   "dropout": 0.0,
-  "embed_dim_tokenizer": 512,
   "embed_dim_world_model": 256,
   "embed_pdrop": 0.1,
   "entropy_weight": 0.001,
   "eos_token_id": 50256,
   "gamma": 0.995,
   "grad_acc_steps_actor_critic": 1,
-  "grad_acc_steps_tokenizer": 1,
   "grad_acc_steps_world_model": 1,
   "imagine_horizon_train_actor_critic": 20,
   "in_channels": 3,

     1,
     1
   ],
+  "device": "cuda:0",
   "dropout": 0.0,
+  "embed_dim_discrete_autoencoder": 512,
   "embed_dim_world_model": 256,
   "embed_pdrop": 0.1,
   "entropy_weight": 0.001,
   "eos_token_id": 50256,
   "gamma": 0.995,
   "grad_acc_steps_actor_critic": 1,
+  "grad_acc_steps_discrete_autoencoder": 1,
   "grad_acc_steps_world_model": 1,
   "imagine_horizon_train_actor_critic": 20,
   "in_channels": 3,