fix(typo): class name SanmDecoer => SanmDecoder (#2110)

* fix(type): fix class name

* fix(typo): fix class name

Author: xingchensong

wenet/paraformer/export_jit.py

@@ -5,7 +5,7 @@
 import torch
 import yaml
 from wenet.paraformer.cif import Cif
-from wenet.paraformer.layers import (SanmDecoer, SanmEncoder)
+from wenet.paraformer.layers import (SanmDecoder, SanmEncoder)
 from wenet.paraformer.paraformer import Paraformer
 from wenet.transformer.cmvn import GlobalCMVN
 from wenet.utils.checkpoint import load_checkpoint
@@ -33,9 +33,9 @@ def init_model(configs):
     encoder = SanmEncoder(global_cmvn=global_cmvn,
                           input_size=configs['lfr_conf']['lfr_m'] * input_dim,
                           **configs['encoder_conf'])
-    decoder = decoder = SanmDecoer(vocab_size=vocab_size,
-                                   encoder_output_size=encoder.output_size(),
-                                   **configs['decoder_conf'])
+    decoder = decoder = SanmDecoder(vocab_size=vocab_size,
+                                    encoder_output_size=encoder.output_size(),
+                                    **configs['decoder_conf'])
     predictor = Cif(**configs['cif_predictor_conf'])
     model = Paraformer(
         encoder=encoder,

wenet/paraformer/layers.py

@@ -372,7 +372,7 @@ def forward(
         return x, tgt_mask, memory, memory_mask
 
 
-class SanmDecoer(TransformerDecoder):
+class SanmDecoder(TransformerDecoder):
 
     def __init__(
         self,

wenet/paraformer/paraformer.py

@@ -22,7 +22,7 @@
 import torch
 from wenet.paraformer.cif import Cif
 
-from wenet.paraformer.layers import SanmDecoer, SanmEncoder
+from wenet.paraformer.layers import SanmDecoder, SanmEncoder
 from wenet.paraformer.layers import LFR
 from wenet.paraformer.search import (paraformer_beam_search,
                                      paraformer_greedy_search)
@@ -37,7 +37,7 @@ class Paraformer(torch.nn.Module):
 
     """
 
-    def __init__(self, encoder: SanmEncoder, decoder: SanmDecoer,
+    def __init__(self, encoder: SanmEncoder, decoder: SanmDecoder,
                  predictor: Cif):
 
         super().__init__()