MobileNetV1_multilabel.yaml
parent
e1502ac353
commit
2c7baf248f
|
@ -122,8 +122,8 @@ Infer:
|
|||
|
||||
Metric:
|
||||
Train:
|
||||
- HammingDistance:
|
||||
- AccuracyScore:
|
||||
- HammingDistance:
|
||||
Eval:
|
||||
- HammingDistance:
|
||||
- AccuracyScore:
|
||||
- HammingDistance:
|
||||
|
|
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue