Dev gzf train (#1524)

* trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * train (#1521) (#1522) * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer * trainer
2025-09-15 14:48:36 +08:00 · 2024-03-21 12:52:02 +08:00 · 2024-03-21 12:52:02 +08:00 · cd001acf68
commit cd001acf68
parent 4482bbcbb9
2 changed files with 3 additions and 2 deletions
--- a/funasr/models/paraformer/decoder.py
+++ b/funasr/models/paraformer/decoder.py
@ -351,9 +351,9 @@ class ParaformerSANMDecoder(BaseTransformerDecoder):
        hlens: torch.Tensor,
        ys_in_pad: torch.Tensor,
        ys_in_lens: torch.Tensor,
-        return_hidden: bool = False,
-        return_both: bool= False,
        chunk_mask: torch.Tensor = None,
+        return_hidden: bool = False,
+        return_both: bool = False,
    ) -> Tuple[torch.Tensor, torch.Tensor]:
        """Forward decoder.

--- a/funasr/utils/misc.py
+++ b/funasr/utils/misc.py
@ -1,6 +1,7 @@
 import os
 import io
 import shutil
+import logging
 from collections import OrderedDict
 import numpy as np
 from omegaconf import DictConfig, OmegaConf