diff --git a/egs/aishell/ASR/transformer_ctc/.train.py.swp b/egs/aishell/ASR/transformer_ctc/.train.py.swp
index 788c36d88..09b705645 100644
Binary files a/egs/aishell/ASR/transformer_ctc/.train.py.swp and b/egs/aishell/ASR/transformer_ctc/.train.py.swp differ
diff --git a/egs/aishell/ASR/transformer_ctc/train.py b/egs/aishell/ASR/transformer_ctc/train.py
index 6600b0b3d..de736b00e 100755
--- a/egs/aishell/ASR/transformer_ctc/train.py
+++ b/egs/aishell/ASR/transformer_ctc/train.py
@@ -491,6 +491,7 @@ def train_one_epoch(
     valid_dl: torch.utils.data.DataLoader,
     tb_writer: Optional[SummaryWriter] = None,
     world_size: int = 1,
+    wb = None,
 ) -> None:
     """Train the model for one epoch.