在很多pytorch写的人工智能项目中,都能看到这样一句:parser = argparse.ArgumentParser(),本文就简单记录下它们的使用方法

部分内容参考:

https://blog.csdn.net/lizhiyuanbest/article/details/104975848

方式1—单个参数

创建a.py

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("square", help="display a square of a given number", type=int)
args = parser.parse_args()

s = args.square ** 2
print(s)

命令行执行:

$ python a.py 2
4

方式2—多个参数

创建b.py

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("square", help="display a square of a given number", type=int)
parser.add_argument("add1", help="display a sum of two given numbers", type=int)
parser.add_argument("add2", help="display a sum of two given numbers", type=int)
args = parser.parse_args()

s = args.square ** 2
a = args.add1 + args.add2
print(s)
print(a)

命令行执行结果:

$ python b.py 2 10 20
4
30

可见是将三个整数按顺序赋给三个参数

方式3—乱序赋值

创建c.py

import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--square", help="display a square of a given number", type=int)
parser.add_argument("--add1", help="display a sum of two given numbers", type=int)
parser.add_argument("--add2", help="display a sum of two given numbers", type=int)
args = parser.parse_args()

s = args.square ** 2
a = args.add1 + args.add2
print(s)
print(a)

⚠️注意参数前面有--

命令行运行方式1:

$ python c.py --square 2 --add1 10 --add2 20
4
30

命令行运行方式2(打乱顺序):

$ python c.py --add1 2 --add2 10 --square 20
400
12

敲pytorch的同学掌握这些应该就足够使用了

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐