From f91c981b2c60661db347363ba20a701aeed3eb33 Mon Sep 17 00:00:00 2001
From: NolanB <nboukachab@teklia.com>
Date: Wed, 17 May 2023 15:54:58 +0200
Subject: [PATCH] remove teacher_forcing_error_rate
---
dan/manager/training.py | 9 +--------
1 file changed, 1 insertion(+), 8 deletions(-)
diff --git a/dan/manager/training.py b/dan/manager/training.py
index cb0673f0..eebe6c6e 100644
--- a/dan/manager/training.py
+++ b/dan/manager/training.py
@@ -1086,14 +1086,7 @@ class Manager(OCRManager):
reduced_size = [s[:2] for s in batch_data["imgs_reduced_shape"]]
y_len = batch_data["labels_len"]
- # add errors in teacher forcing
- if (
- "teacher_forcing_error_rate" in self.params["training_params"]
- and self.params["training_params"]["teacher_forcing_error_rate"] is not None
- ):
- error_rate = self.params["training_params"]["teacher_forcing_error_rate"]
- simulated_y_pred, y_len = self.add_label_noise(y, y_len, error_rate)
- elif "label_noise_scheduler" in self.params["training_params"]:
+ if "label_noise_scheduler" in self.params["training_params"]:
error_rate = (
self.params["training_params"]["label_noise_scheduler"][
"min_error_rate"
--
GitLab