1、argparse框架中的函数add_argument()
parser=argparse.ArgumentParser()
parser.add_argument('--gpu',action='store_true',help='whether use gpu')
args=parser.parse_args()
注意:
add_argument()函数中的action参数,当调用'gpu'参数时,args.gpu才为True,因此一开始最好添加一个参数default=True
即:
parser.add_argument('--gpu',action='store_true',default=True,help='whether use gpu')
2、LSTM模型参数中,数据类型都是整型,最好不要传进float类型参数。