diff --git a/dan/manager/metrics.py b/dan/manager/metrics.py index 8c4682fed470745d9604d68a0f9902bd695a9735..ecb94a86635c91d5b56bf7e6bbba440e4052b9b0 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 042c53a73674100b6efaba097cf2e5a485bcf5af..5821d2e804f3129a263f75da0723c3c5888c97af 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",