Fix an error in DDP training.

2025-08-09 10:02:22 +00:00 · 2021-07-25 22:33:09 +08:00 · 2021-07-25 22:33:09 +08:00 · 78bb65ed78
commit 78bb65ed78
parent 8055bf31a0
1 changed files with 2 additions and 0 deletions
--- a/egs/librispeech/ASR/tdnn_lstm_ctc/train.py
+++ b/egs/librispeech/ASR/tdnn_lstm_ctc/train.py
@ -214,6 +214,8 @@ def save_checkpoint(
      model:
        The training model.
    """
+    if rank != 0:
+        return
    filename = params.exp_dir / f"epoch-{params.cur_epoch}.pt"
    save_checkpoint_impl(
        filename=filename,