Made the default values sane
This commit is contained in:
parent
bd9fad4617
commit
070949d425
2 changed files with 5 additions and 5 deletions
4
args.py
4
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',
|
||||
|
|
|
@ -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,
|
||||
|
|
Loading…
Reference in a new issue