mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 10:02:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix an error in DDP training.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-07-25 22:33:09 +08:00
parent 8055bf31a0
commit 78bb65ed78
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/tdnn_lstm_ctc/train.py
View File

@ -214,6 +214,8 @@ def save_checkpoint(
model: model:
The training model. The training model.
""" """
if rank != 0:
return
filename = params.exp_dir / f"epoch-{params.cur_epoch}.pt" filename = params.exp_dir / f"epoch-{params.cur_epoch}.pt"
save_checkpoint_impl( save_checkpoint_impl(
filename=filename, filename=filename,