-
Notifications
You must be signed in to change notification settings - Fork 158
Expand file tree
/
Copy patharguments.py
More file actions
27 lines (21 loc) · 721 Bytes
/
Copy patharguments.py
File metadata and controls
27 lines (21 loc) · 721 Bytes
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
"""
This file contains the arguments to parse at command line.
File main.py will call get_args, which then the arguments
will be returned.
"""
import argparse
def get_args():
"""
Description:
Parses arguments at command line.
Parameters:
None
Return:
args - the arguments parsed
"""
parser = argparse.ArgumentParser()
parser.add_argument('--mode', dest='mode', type=str, default='train') # can be 'train' or 'test'
parser.add_argument('--actor_model', dest='actor_model', type=str, default='') # your actor model filename
parser.add_argument('--critic_model', dest='critic_model', type=str, default='') # your critic model filename
args = parser.parse_args()
return args