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
2026-10-12 02:19:18 +00:00 · 2024-03-21 12:52:02 +08:00
parent 4482bbcbb9
commit cd001acf68
2 changed files with 3 additions and 2 deletions
@@ -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.

@@ -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