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

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

Browse Source
This commit is contained in:
di.wu 2023-10-07 15:46:22 +08:00
parent 6ae98ef111
commit d618951e87
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
wekws/bin/train.py
View File

@ -141,14 +141,13 @@ 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)
num_params = count_parameters(model)
print('the number of model params: {}'.format(num_params))