From b71481db63b1b71deb8c34e5142f239bd1cd50c8 Mon Sep 17 00:00:00 2001
From: NolanB <nboukachab@teklia.com>
Date: Wed, 17 May 2023 13:04:07 +0200
Subject: [PATCH] remove syn_max_lines and syn_prob_lines
---
dan/manager/metrics.py | 2 --
dan/manager/training.py | 6 ------
dan/ocr/document/train.py | 2 --
3 files changed, 10 deletions(-)
diff --git a/dan/manager/metrics.py b/dan/manager/metrics.py
index a36ecc1a..ecb94a86 100644
--- a/dan/manager/metrics.py
+++ b/dan/manager/metrics.py
@@ -223,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 df33eda3..83a3cd56 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