Pneuma/pneuma-pygame
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Made the default values sane

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

4
args.py
View file

@ -23,12 +23,12 @@ def parse_args():
# Define episodes and agents # Define episodes and agents
parser.add_argument('--n_episodes', parser.add_argument('--n_episodes',
type=int, type=int,
default=300, default=12000,
help="Number of episodes.") help="Number of episodes.")
parser.add_argument('--ep_length', parser.add_argument('--ep_length',
type=int, type=int,
default=5000, default=700,
help="Length of each episode.") help="Length of each episode.")
parser.add_argument('--n_agents', parser.add_argument('--n_agents',

6
utils/hyperparams.py
View file

@ -1,8 +1,8 @@
HPARAMS = { HPARAMS = {
"horizon": 2048, "horizon": 256,
"num_epochs": 15, "num_epochs": 10,
"batch_size": 128, "batch_size": 15,
"policy_clip": 0.1, "policy_clip": 0.1,
"discount_factor": 0.99, "discount_factor": 0.99,