diff --git a/ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml b/ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml index 6838710c326af0d9a549a3340ffd1326d784dced..969d8161f2f3d9bb60f7afe9bd1d33ff155a6d95 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: - Eval: - HammingDistance: + Eval: - AccuracyScore: + - HammingDistance: diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index 884a05bb141947d70d2a20c2d88967bcbe6626ea..a8b3c90faff659a7c7b6ae8af426fd7e6be5c1ef 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,