瀏覽代碼

show eval_metric on visualdl

will-jl944 4 年之前
父節點
當前提交
3934144f7b
共有 1 個文件被更改,包括 8 次插入0 次删除
  1. 8 0
      dygraph/paddlex/cv/models/base.py

+ 8 - 0
dygraph/paddlex/cv/models/base.py

@@ -396,6 +396,14 @@ class BaseModel:
                     # 保存最优模型
                     if local_rank == 0:
                         self.eval_metrics, self.eval_details = eval_result
+                        if use_vdl:
+                            for k, v in self.eval_metrics.items():
+                                try:
+                                    log_writer.add_scalar(
+                                        '{}-Metrics/Eval(Epoch): {}'.format(
+                                            task_id, k), v, i + 1)
+                                except TypeError:
+                                    pass
                         logging.info('[EVAL] Finished, Epoch={}, {} .'.format(
                             i + 1, dict2str(self.eval_metrics)))
                         best_accuracy_key = list(self.eval_metrics.keys())[0]