default params change

corl-team · Dec 4, 2023 · 007762b · 007762b
1 parent 6ce8f74
commit 007762b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/train_meta_standalone.ipynb b/examples/train_meta_standalone.ipynb
@@ -393,11 +393,11 @@
     "    benchmark_id: str = \"Trivial\"\n",
     "    # agent\n",
     "    action_emb_dim: int = 16\n",
-    "    rnn_hidden_dim: int = 64\n",
+    "    rnn_hidden_dim: int = 1024\n",
     "    rnn_num_layers: int = 1\n",
-    "    head_hidden_dim: int = 64\n",
+    "    head_hidden_dim: int = 256\n",
     "    # training\n",
-    "    num_envs: int = 4096\n",
+    "    num_envs: int = 8192\n",
     "    num_steps_per_env: int = 4096\n",
     "    num_steps_per_update: int = 32\n",
     "    update_epochs: int = 4\n",