From 070949d425f9b66a16745230c61aea8939a2cecf Mon Sep 17 00:00:00 2001
From: Vasilis Valatsos <vasilvalat@gmail.com>
Date: Fri, 22 Mar 2024 13:20:01 +0100
Subject: [PATCH] Made the default values sane

---
 args.py              | 4 ++--
 utils/hyperparams.py | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/args.py b/args.py
index 3e26d88..53fde64 100644
--- a/args.py
+++ b/args.py
@@ -23,12 +23,12 @@ def parse_args():
     # Define episodes and agents
     parser.add_argument('--n_episodes',
                         type=int,
-                        default=300,
+                        default=12000,
                         help="Number of episodes.")
 
     parser.add_argument('--ep_length',
                         type=int,
-                        default=5000,
+                        default=700,
                         help="Length of each episode.")
 
     parser.add_argument('--n_agents',
diff --git a/utils/hyperparams.py b/utils/hyperparams.py
index 2f5a85e..0cd7274 100644
--- a/utils/hyperparams.py
+++ b/utils/hyperparams.py
@@ -1,8 +1,8 @@
 HPARAMS = {
 
-    "horizon": 2048,
-    "num_epochs": 15,
-    "batch_size": 128,
+    "horizon": 256,
+    "num_epochs": 10,
+    "batch_size": 15,
 
     "policy_clip": 0.1,
     "discount_factor": 0.99,