Made the default values sane

This commit is contained in:
Vasilis Valatsos 2024-03-22 13:20:01 +01:00
parent bd9fad4617
commit 070949d425
2 changed files with 5 additions and 5 deletions

View file

@ -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',

View file

@ -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,