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,