ruffy369
/

iris-pong

Reinforcement Learning

Inference Endpoints

Model card Files Files and versions Community

ruffy369 commited on Jul 25

Commit

e371cc0

•

1 Parent(s): 04653e0

Update config.json

Files changed (1) hide show

config.json +4 -3

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,
@@ -45,7 +46,7 @@
   "resolution": 64,
   "sequence_length": 20,
   "tokens_per_block": 17,
-  "transformers_version": "4.42.0.dev0",
   "use_cache": true,
   "use_original_obs_actor_critic": false,
   "vocab_size": 512,

     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,
   "resolution": 64,
   "sequence_length": 20,
   "tokens_per_block": 17,
+  "transformers_version": "4.44.0.dev0",
   "use_cache": true,
   "use_original_obs_actor_critic": false,
   "vocab_size": 512,