From 9870933a575458afaea14fbe9f711a68f4085320 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Thu, 11 May 2023 19:18:18 +0200
Subject: [PATCH] only update metrics module for now
---
dan/manager/metrics.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/dan/manager/metrics.py b/dan/manager/metrics.py
index a9791c39..fdcc0ea5 100644
--- a/dan/manager/metrics.py
+++ b/dan/manager/metrics.py
@@ -140,6 +140,8 @@ class MetricManager:
"loss",
"loss_ctc",
"loss_ce",
+ "syn_max_lines",
+ "syn_prob_lines",
]:
value = np.average(
self.epoch_metrics[metric_name],
@@ -208,6 +210,8 @@ class MetricManager:
"loss_ctc",
"loss_ce",
"loss",
+ "syn_max_lines",
+ "syn_prob_lines",
]:
metrics[metric_name] = [
values[metric_name],
--
GitLab