diff --git a/dan/manager/metrics.py b/dan/manager/metrics.py index a36ecc1a1543d89f742aac5c4cdf3f859f798223..ecb94a86635c91d5b56bf7e6bbba440e4052b9b0 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 df33eda3b58bd1e9ba2c81ef966f2343c3cc5e0d..83a3cd561fd3249ed13757e4b5270180d2e69471 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 1d6cfb9a48603288f2e3dea8eab9e9f5ce7de0de..5a5ebd006e166f1f3c7b872bd2a68dc7b9cbe10a 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",