方式一:使用sys模块
import sys a = sys.argv[0] b = sys.argv[1] c = sys.argv[2] print("a=%s,b=%s,c=%s" % (a,b,c)) # 执行python 脚本名 参数一 参数二 a 就是脚本名、b是第一个参数、c是第二个参数
方式二:
import argparse parser = argparse.ArgumentParser(description=‘manual to this script‘) parser.add_argument("--gpus", type=str, default="0") #添加参数 --gpus parser.add_argument("--batch-size", type=int, default=33) #添加参数 --batch-size args = parser.parse_args() print(args.gpus) print(args.batch_size) #执行方式 python 脚本名 --gpus=参数 --batch-size=参数
原文:https://www.cnblogs.com/yanhonghong/p/11644830.html