diff --git a/dan/manager/metrics.py b/dan/manager/metrics.py index a9791c39ae993041e3884ecce178508963a0cb91..fdcc0ea53dd63e89b44ddaa17c295c68ea9938e5 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],