From fb176047c3aeba1611107fbb38513e12196d235c Mon Sep 17 00:00:00 2001 From: Mddct Date: Mon, 8 Jan 2024 20:39:04 +0800 Subject: [PATCH] [paraformer] make length_normalized_loss false --- wenet/paraformer/convert_paraformer_to_wenet_config_and_ckpt.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/wenet/paraformer/convert_paraformer_to_wenet_config_and_ckpt.py b/wenet/paraformer/convert_paraformer_to_wenet_config_and_ckpt.py index bf5c7254c..f60b83e5c 100644 --- 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()