zhangyubo0722 1 rok pred
rodič
commit
e98f14e842

+ 2 - 1
paddlex/repo_apis/PaddleClas_api/cls/runner.py

@@ -190,12 +190,13 @@ def _extract_eval_metrics(stdout: str) -> dict:
 
     _DP = r"[-+]?[0-9]*\.?[0-9]+(?:[eE][-+]?[0-9]+)?"
     patterns = [
+        r"\[Eval\]\[Epoch 0\]\[Avg\].*top1: (_dp)".replace("_dp", _DP),
         r"\[Eval\]\[Epoch 0\]\[Avg\].*top1: (_dp), top5: (_dp)".replace("_dp", _DP),
         r"\[Eval\]\[Epoch 0\]\[Avg\].*recall1: (_dp), recall5: (_dp), mAP: (_dp)".replace(
             "_dp", _DP
         ),
     ]
-    keys = [["val.top1", "val.top5"], ["recall1", "recall5", "mAP"]]
+    keys = [["val.top1"], ["val.top1", "val.top5"], ["recall1", "recall5", "mAP"]]
 
     metric_dict = dict()
     for pattern, key in zip(patterns, keys):