From 4d63e07430c3b97608dbfe0179054c792bfaeea7 Mon Sep 17 00:00:00 2001
From: Nolan <nboukachab@teklia.com>
Date: Wed, 24 May 2023 14:48:15 +0000
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 7e02174b..876e788c 100644
--- a/dan/manager/training.py
+++ b/dan/manager/training.py
@@ -1087,14 +1087,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