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