From 007762bd7cd9190bcf6a9c7c21bb2a2ffab7753f Mon Sep 17 00:00:00 2001 From: Howuhh Date: Mon, 4 Dec 2023 14:10:43 +0300 Subject: [PATCH] default params change --- examples/train_meta_standalone.ipynb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/train_meta_standalone.ipynb b/examples/train_meta_standalone.ipynb index 6720ce1..cf14465 100644 --- 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",