diff --git a/ppcls/configs/Cartoonface/ResNet50_icartoon.yaml b/ppcls/configs/Cartoonface/ResNet50_icartoon.yaml index 01ab83e17..bd39afd90 100644 --- a/ppcls/configs/Cartoonface/ResNet50_icartoon.yaml +++ b/ppcls/configs/Cartoonface/ResNet50_icartoon.yaml @@ -14,7 +14,7 @@ Global: use_visualdl: False image_shape: [3, 224, 224] infer_imgs: - save_inference_dir: "./output" + save_inference_dir: "./inference" feature_normalize: True Arch: diff --git a/ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml b/ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml index 5c8a01ab2..9a8469d4b 100644 --- a/ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml +++ b/ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml @@ -19,6 +19,9 @@ Global: # model architecture Arch: name: RecModel + infer_output_key: features + infer_add_softmax: False + Backbone: name: ResNet50_vd pretrained: True