维石
2024-05-28 64d703207304a062241f19eb19f6e3e7e8ab4399
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