From ebbcf87d6ce0eb50ab1faf8df563573b2732ba34 Mon Sep 17 00:00:00 2001
From: Nolan <nboukachab@teklia.com>
Date: Tue, 23 May 2023 12:51:06 +0000
Subject: [PATCH] Remove syn_max_lines, syn_prob_lines metrics
---
dan/manager/metrics.py | 4 ----
dan/manager/training.py | 6 ------
dan/ocr/document/train.py | 2 --
3 files changed, 12 deletions(-)
diff --git a/dan/manager/metrics.py b/dan/manager/metrics.py
index 8c4682fe..ecb94a86 100644
--- a/dan/manager/metrics.py
+++ b/dan/manager/metrics.py
@@ -152,8 +152,6 @@ class MetricManager:
"loss",
"loss_ctc",
"loss_ce",
- "syn_max_lines",
- "syn_prob_lines",
]:
value = float(
np.average(
@@ -225,8 +223,6 @@ class MetricManager:
"loss_ctc",
"loss_ce",
"loss",
- "syn_max_lines",
- "syn_prob_lines",
]:
metrics[metric_name] = [
values[metric_name],
diff --git a/dan/manager/training.py b/dan/manager/training.py
index 042c53a7..5821d2e8 100644
--- a/dan/manager/training.py
+++ b/dan/manager/training.py
@@ -1188,12 +1188,6 @@ class Manager(OCRManager):
"str_x": str_x,
"loss": sum_loss.item(),
"loss_ce": loss_ce.item(),
- "syn_max_lines": self.dataset.train_dataset.get_syn_max_lines()
- if self.params["dataset_params"]["config"]["synthetic_data"]
- else 0,
- "syn_prob_lines": self.dataset.train_dataset.get_syn_proba_lines()
- if self.params["dataset_params"]["config"]["synthetic_data"]
- else 0,
}
return values
diff --git a/dan/ocr/document/train.py b/dan/ocr/document/train.py
index 1d6cfb9a..5a5ebd00 100644
--- a/dan/ocr/document/train.py
+++ b/dan/ocr/document/train.py
@@ -212,8 +212,6 @@ def get_config():
"cer",
"wer",
"wer_no_punct",
- "syn_max_lines",
- "syn_prob_lines",
], # Metrics name for training
"eval_metrics": [
"cer",
--
GitLab