From 78bb65ed783cc79c25010d75a1619cb809d4d649 Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Sun, 25 Jul 2021 22:33:09 +0800
Subject: [PATCH] Fix an error in DDP training.

---
 egs/librispeech/ASR/tdnn_lstm_ctc/train.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/egs/librispeech/ASR/tdnn_lstm_ctc/train.py b/egs/librispeech/ASR/tdnn_lstm_ctc/train.py
index 78f276a13..9ce4f69da 100755
--- 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,