zhifu gao
2024-05-15 a0f03bd2a87d97d47a1636bbe6f0855a43160331
funasr/models/ctc/ctc.py
@@ -40,9 +40,7 @@
                logging.warning("ignore_nan_grad option is not supported for warp_ctc")
            self.ctc_loss = warp_ctc.CTCLoss(size_average=True, reduce=reduce)
        else:
            raise ValueError(
                f'ctc_type must be "builtin" or "warpctc": {self.ctc_type}'
            )
            raise ValueError(f'ctc_type must be "builtin" or "warpctc": {self.ctc_type}')
        self.reduce = reduce