def parse_args():
import argparse
parser = argparse.ArgumentParser(description='tensor analysis')
parser.add_argument('-i',
'--input',
default='.',
help='path of your datas')
parser.add_argument('-n',
'--number',
default=50,
type=int,
help='number of result you want to see')
parser.add_argument('-o',
'--output',
default='.',
help='path where to save')
args = parser.parse_args()
return args
def main(args):
is_bfloat = IsBf16()
maybe_print("is_bfloat {}".format(is_bfloat))
in_f0 = open(args.input, "rb")
in_c0 = in_f0.read()
if __name__ == "__main__":
args = parse_args()
main(args)
$ python xxxx.py --input xxxx
python 命令行传参
最新推荐文章于 2024-10-05 08:57:27 发布