MobileNetV1_multilabel.yaml

pull/2180/head
cuicheng01 2022-07-18 05:16:05 +00:00
parent e1502ac353
commit 2c7baf248f
2 changed files with 3 additions and 3 deletions

View File

@ -122,8 +122,8 @@ Infer:
Metric:
Train:
- HammingDistance:
- AccuracyScore:
- HammingDistance:
Eval:
- HammingDistance:
- AccuracyScore:
- HammingDistance:

View File

@ -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,