AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[wekws] fix log (only one process print model) (#150)

Browse Source 
Co-authored-by: di.wu <di.wu@diwudeMacBook-Pro.local>
This commit is contained in:
Di Wu 2023-10-09 17:33:52 +08:00 committed by GitHub
parent 6ae98ef111
commit 58859d580d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
wekws/bin/train.py
View File

@ -141,15 +141,14 @@ def main():
configs['model']['cmvn'] = {}
configs['model']['cmvn']['norm_var'] = args.norm_var
configs['model']['cmvn']['cmvn_file'] = args.cmvn_file
# Init asr model from configs
model = init_model(configs['model'])
if rank == 0:
saved_config_path = os.path.join(args.model_dir, 'config.yaml')
with open(saved_config_path, 'w') as fout:
data = yaml.dump(configs)
fout.write(data)
# Init asr model from configs
model = init_model(configs['model'])
print(model)
print(model)
num_params = count_parameters(model)
print('the number of model params: {}'.format(num_params))