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