From 2c7baf248f355ce5b3d7ab95128b14992dd2dbfc Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Mon, 18 Jul 2022 05:16:05 +0000 Subject: [PATCH] MobileNetV1_multilabel.yaml --- .../quick_start/professional/MobileNetV1_multilabel.yaml | 4 ++-- ppcls/engine/engine.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml b/ppcls/configs/quick_start/professional/MobileNetV1_multilabel.yaml index 6838710c..969d8161 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 884a05bb..a8b3c90f 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, -- GitLab