aky15
2023-03-16 3e333c0abf31825e84d9673faf5e77601ced1112
funasr/models_transducer/espnet_transducer_model_unified.py
@@ -455,7 +455,8 @@
                gather=True,
        )
        if not self.training and (self.report_cer or self.report_wer):
        #if not self.training and (self.report_cer or self.report_wer):
        if self.report_cer or self.report_wer:
            if self.error_calculator is None:
                self.error_calculator = ErrorCalculator(
                    self.decoder,
@@ -468,7 +469,6 @@
                )
            cer_transducer, wer_transducer = self.error_calculator(encoder_out, target)
            return loss_transducer, cer_transducer, wer_transducer
        return loss_transducer, None, None