fix code style

pull/3804/head
warrentdrew 1 year ago
parent 0b16fbc89a
commit fb27a235db

@ -591,7 +591,7 @@ def get_predictor(
config = inference.Config( config = inference.Config(
str(Path(model_dir) / model_file), str(Path(model_dir) / params_file)) str(Path(model_dir) / model_file), str(Path(model_dir) / params_file))
if device != "npu": if paddle.__version__ <= "2.5.2" and paddle.__version__ != "0.0.0":
config.enable_memory_optim() config.enable_memory_optim()
config.switch_ir_optim(True) config.switch_ir_optim(True)
if device == "gpu": if device == "gpu":

@ -219,18 +219,18 @@ def parse_args():
) )
# other # other
parser.add_argument( 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( parser.add_argument(
"--nxpu", "--nxpu",
type=int, type=int,
default=0, 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( parser.add_argument(
"--nnpu", "--nnpu",
type=int, type=int,
default=0, default=0,
help="if wish to use npu, set ngpu == 0 and nxpu == 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("--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.")

Loading…
Cancel
Save