fix help message

pull/3804/head
warrentdrew 1 year ago
parent b8dace46e3
commit ac508e78f9

@ -259,7 +259,7 @@ def main():
elif args.ngpu == 0 and args.nxpu == 0 and args.nnpu == 0:
paddle.set_device("cpu")
else:
print("ngpu or nxpu or nnpu should >= 0 !")
print("ngpu, nxpu and nnpu should be >= 0")
evaluate(args)

@ -299,18 +299,21 @@ def parse_args():
default=None,
help="dir to save inference models")
parser.add_argument(
"--ngpu", type=int, default=1, help="if ngpu == 0, use cpu or xpu or npu.")
"--ngpu",
type=int,
default=1,
help="if wish to use gpu, set ngpu > 0, otherwise use xpu, npu or cpu.")
parser.add_argument(
"--nxpu",
type=int,
default=0,
help="if wish to use xpu, set ngpu == 0 and nxpu > 0, and if ngpu == 0 and nxpu == 0 and nnpu == 0, use cpu."
help="if wish to use xpu, set ngpu == 0 and nxpu > 0, otherwise use gpu, npu or cpu."
)
parser.add_argument(
"--nnpu",
type=int,
default=0,
help="if wish to use npu, set ngpu == 0 and nnpu > 0, and if ngpu == 0 and nxpu == 0 and nnpu == 0, use cpu."
help="if wish to use npu, set ngpu == 0 and nnpu > 0, otherwise use gpu, xpu or cpu."
)
parser.add_argument(
"--text",
@ -350,7 +353,7 @@ def main():
elif args.ngpu == 0 and args.nxpu == 0 or args.nnpu == 0:
paddle.set_device("cpu")
else:
print("ngpu or nxpu or nnpu should >= 0 !")
print("ngpu, nxpu and nnpu should be >= 0")
evaluate(args)

Loading…
Cancel
Save