提交 538ca47e 编写于 作者: D dongshuilong

fix levit384 inference size bug

上级 2135fa85
...@@ -6,7 +6,7 @@ gpu_list:0|0,1 ...@@ -6,7 +6,7 @@ gpu_list:0|0,1
-o Global.auto_cast:null -o Global.auto_cast:null
-o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120 -o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
-o Global.output_dir:./output/ -o Global.output_dir:./output/
-o DataLoader.Train.sampler.batch_size:8 -o DataLoader.Train.sampler.batch_size:2
-o Global.pretrained_model:null -o Global.pretrained_model:null
train_model_name:latest train_model_name:latest
train_infer_img_dir:./dataset/ILSVRC2012/val train_infer_img_dir:./dataset/ILSVRC2012/val
...@@ -37,7 +37,7 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/L ...@@ -37,7 +37,7 @@ pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/L
infer_model:../inference/ infer_model:../inference/
infer_export:True infer_export:True
infer_quant:Fasle infer_quant:Fasle
inference:python/predict_cls.py -c configs/inference_cls.yaml -o PreProcess.transform_ops.0.ResizeImage.resize_short=384 -o PreProcess.transform_ops.1.CropImage.size=384 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.use_gpu:True|False
-o Global.enable_mkldnn:True|False -o Global.enable_mkldnn:True|False
-o Global.cpu_num_threads:1|6 -o Global.cpu_num_threads:1|6
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册