[paraformer] make length_normalized_loss false

wenet-e2e · Jan 8, 2024 · fb17604 · fb17604
1 parent 14352a4
commit fb17604
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/wenet/paraformer/convert_paraformer_to_wenet_config_and_ckpt.py b/wenet/paraformer/convert_paraformer_to_wenet_config_and_ckpt.py
@@ -156,6 +156,8 @@ def convert_to_wenet_yaml(configs, wenet_yaml_path: str,
     configs['max_epoch'] = 100
     configs['log_interval'] = 100
 
+    configs['model_conf']['length_normalized_loss'] = False
+
     with open(wenet_yaml_path, '+w') as f:
         f.write(yaml.dump(configs))
         f.flush()