28 lines
721 B
Python
28 lines
721 B
Python
|
"""
|
||
|
|
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
|