diff --git a/test_tipc/configs/ConvNeXt/ConvNeXt_tiny_train_infer_python.txt b/test_tipc/configs/ConvNeXt/ConvNeXt_tiny_train_infer_python.txt index 11b4007ef9fbaef563b028a10bf9f42eb2581f94..9a683deca89dd9e46a851c65fdafc5df06dd27a8 100644 --- a/test_tipc/configs/ConvNeXt/ConvNeXt_tiny_train_infer_python.txt +++ b/test_tipc/configs/ConvNeXt/ConvNeXt_tiny_train_infer_python.txt @@ -39,10 +39,10 @@ infer_export:True infer_quant:Fasle inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=256 -o PreProcess.transform_ops.1.CropImage.size=224 -o Global.use_gpu:True|False --o Global.enable_mkldnn:True|False +-o Global.enable_mkldnn:False -o Global.cpu_num_threads:1|6 -o Global.batch_size:1|16 --o Global.use_tensorrt:True|False +-o Global.use_tensorrt:False -o Global.use_fp16:True|False -o Global.inference_model_dir:../inference -o Global.infer_imgs:../dataset/ILSVRC2012/val @@ -51,4 +51,4 @@ inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.tran null:null null:null ===========================infer_benchmark_params========================== -random_infer_input:[{float32,[3,224,224]}] \ No newline at end of file +random_infer_input:[{float32,[3,224,224]}]