提交 2c7baf24 编写于 作者: C cuicheng01

MobileNetV1_multilabel.yaml

上级 e1502ac3
...@@ -122,8 +122,8 @@ Infer: ...@@ -122,8 +122,8 @@ Infer:
Metric: Metric:
Train: Train:
- HammingDistance:
- AccuracyScore: - AccuracyScore:
Eval:
- HammingDistance: - HammingDistance:
Eval:
- AccuracyScore: - AccuracyScore:
- HammingDistance:
...@@ -456,7 +456,7 @@ class Engine(object): ...@@ -456,7 +456,7 @@ class Engine(object):
assert self.mode == "export" assert self.mode == "export"
use_multilabel = self.config["Global"].get( use_multilabel = self.config["Global"].get(
"use_multilabel", "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) model = ExportModel(self.config["Arch"], self.model, use_multilabel)
if self.config["Global"]["pretrained_model"] is not None: if self.config["Global"]["pretrained_model"] is not None:
load_dygraph_pretrain(model.base_model, load_dygraph_pretrain(model.base_model,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册