[paraformer] fix input dim

wenet-e2e · Jan 8, 2024 · 14352a4 · 14352a4
1 parent 3e2820b
commit 14352a4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/wenet/utils/train_utils.py b/wenet/utils/train_utils.py
@@ -209,6 +209,8 @@ def check_modify_and_save_config(args, configs, symbol_table):
                 'num_mel_bins']
         else:
             input_dim = configs['dataset_conf']['mfcc_conf']['num_mel_bins']
+    else:
+        input_dim = configs['input_dim']
 
     configs, _ = get_blank_id(configs, symbol_table)