diff --git a/dan/manager/training.py b/dan/manager/training.py index 7e02174be28cae62d8319367269445ad9957ed3f..876e788c23c3ec785319535674e3af985790b4a7 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"