diff --git a/ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml b/ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml index 6838710c3..969d8161f 100644 --- a/ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml +++ b/ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml @@ -122,8 +122,8 @@ Infer: Metric: Train: - - HammingDistance: - AccuracyScore: + - HammingDistance: Eval: - - HammingDistance: - AccuracyScore: + - HammingDistance: diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index 884a05bb1..a8b3c90fa 100644 --- a/ppcls/engine/engine.py +++ b/ppcls/engine/engine.py @@ -456,7 +456,7 @@ class Engine(object): assert self.mode == "export" use_multilabel = self.config["Global"].get( "use_multilabel", - False) and "ATTRMetric" in self.config["Metric"]["Eval"][0] + False) or "ATTRMetric" in self.config["Metric"]["Eval"][0] model = ExportModel(self.config["Arch"], self.model, use_multilabel) if self.config["Global"]["pretrained_model"] is not None: load_dygraph_pretrain(model.base_model,