|
|
|
@ -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)
|
|
|
|
|
|
|
|
|
|