diff --git a/tests/config/ResNet50_vd.txt b/tests/config/ResNet50_vd.txt index 9b8f27cc6485ff61f3b3999a870af544afca485b..ed706085985b3f9d405b478d402551a023b26d22 100644 --- a/tests/config/ResNet50_vd.txt +++ b/tests/config/ResNet50_vd.txt @@ -33,7 +33,7 @@ fpgm_export:tools/export_model.py -c ppcls/configs/slim/ResNet50_vd_prune.yaml distill_export:null kl_quant:deploy/slim/quant_post_static.py -c ppcls/configs/ImageNet/ResNet/ResNet50_vd.yaml -o Global.save_inference_dir=./inference export2:null -infer_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/whole_chain/ResNet50_vd_inference.tar +inference_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/whole_chain/ResNet50_vd_inference.tar infer_model:../inference/ infer_export:null infer_quant:Fasle diff --git a/tests/prepare.sh b/tests/prepare.sh index 4c2aa052a57923e3354847c07f2091bb3c0e5666..57e9a949d6bbb2beb8486f42e556e8fd94fe7365 100644 --- a/tests/prepare.sh +++ b/tests/prepare.sh @@ -42,6 +42,9 @@ if [ ${MODE} = "lite_train_infer" ] || [ ${MODE} = "whole_infer" ];then cd ILSVRC2012 mv train.txt train_list.txt mv val.txt val_list.txt + if [ ${MODE} = "lite_train_infer" ];then + cp -r train/* val/ + fi cd ../../ elif [ ${MODE} = "infer" ] || [ ${MODE} = "cpp_infer" ];then # download data