merge main

wenet-e2e · Feb 6, 2025 · a7a5aa5 · a7a5aa5
1 parent 9d50f46
commit a7a5aa5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/wenet/transformer/encoder.py b/wenet/transformer/encoder.py
@@ -106,9 +106,9 @@ def __init__(
 
         assert layer_norm_type in ['layer_norm', 'rms_norm']
         self.normalize_before = normalize_before
+        self.final_norm = final_norm
         self.after_norm = WENET_NORM_CLASSES[layer_norm_type](output_size,
                                                               eps=norm_eps)
-        self.final_norm = final_norm
         self.static_chunk_size = static_chunk_size
         self.use_dynamic_chunk = use_dynamic_chunk
         self.use_dynamic_left_chunk = use_dynamic_left_chunk