|
|
@ -40,6 +40,12 @@ if __name__ == "__main__":
|
|
|
|
"--export_path", type=str, help="path of the jit model to save")
|
|
|
|
"--export_path", type=str, help="path of the jit model to save")
|
|
|
|
parser.add_argument(
|
|
|
|
parser.add_argument(
|
|
|
|
"--model_type", type=str, default='offline', help='offline/online')
|
|
|
|
"--model_type", type=str, default='offline', help='offline/online')
|
|
|
|
|
|
|
|
parser.add_argument(
|
|
|
|
|
|
|
|
'--nxpu',
|
|
|
|
|
|
|
|
type=int,
|
|
|
|
|
|
|
|
default=0,
|
|
|
|
|
|
|
|
choices=[0, 1],
|
|
|
|
|
|
|
|
help="if nxpu == 0 and ngpu == 0, use cpu.")
|
|
|
|
parser.add_argument(
|
|
|
|
parser.add_argument(
|
|
|
|
"--enable-auto-log", action="store_true", help="use auto log")
|
|
|
|
"--enable-auto-log", action="store_true", help="use auto log")
|
|
|
|
args = parser.parse_args()
|
|
|
|
args = parser.parse_args()
|
|
|
|