可以使用argparse库,import后可以开始书写,代码如下:
def get_common_args():
parser = argparse.ArgumentParser()
parser.add_argument('--difficulty', type=str, default='7', help='the difficulty of the game')
args = parser.parse_args()
return args
定义一个函数,首先创建一个解析器,即
parser = argparse.ArgumentParser()
然后加入参数
parser.add_argument("参数名字", type=类型, default='默认值', help='说明')
解析参数
args = parser.parse_args()
然后返回
当需要在创建好以后再添加参数,则可以直接
args.rnn_hidden_dim = 64