-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathrun.py
29 lines (27 loc) · 823 Bytes
/
run.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import argparse
from c2d.runner import Runner
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description="Run an experiment for STEPS total environment steps.")
parser.add_argument(
"--game",
dest="game",
default="breakout",
help="Environment string. Default is breakout.",
)
parser.add_argument(
"--tag",
dest="dtag",
default="unnamed",
help="Training run data string tag. Default is no data storage.",
)
parser.add_argument(
"--iterations",
dest="iterations",
type=int,
default=1,
help="Number of training and evaluation phases. Default is 1 iterations.",
)
args = parser.parse_args()
runner = Runner(args.game, args.iterations, args.dtag)
runner.run()