update description of nxpu

pull/3502/head
USTCKAY 2 years ago
parent 7d1c7cc512
commit 11aa800663

@ -224,7 +224,8 @@ def parse_args():
"--nxpu", "--nxpu",
type=int, type=int,
default=0, default=0,
help="if ngpu == 0 and nxpu == 0, use cpu.") help="if wish to use xpu, set ngpu == 0 and nxpu > 0, and if ngpu == 0 and nxpu == 0, use cpu."
)
parser.add_argument("--test_metadata", type=str, help="test metadata.") parser.add_argument("--test_metadata", type=str, help="test metadata.")
parser.add_argument("--output_dir", type=str, help="output dir.") parser.add_argument("--output_dir", type=str, help="output dir.")
parser.add_argument( parser.add_argument(

@ -304,7 +304,8 @@ def parse_args():
"--nxpu", "--nxpu",
type=int, type=int,
default=0, default=0,
help="if ngpu == 0 and nxpu == 0, use cpu.") help="if wish to use xpu, set ngpu == 0 and nxpu > 0, and if ngpu == 0 and nxpu == 0, use cpu."
)
parser.add_argument( parser.add_argument(
"--text", "--text",
type=str, type=str,

Loading…
Cancel
Save