diff --git a/egs/aishell/ASR/transformer_ctc/.train.py.swp b/egs/aishell/ASR/transformer_ctc/.train.py.swp
index 53c7d3a94..788c36d88 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 a95897ead..6600b0b3d 100755
--- a/egs/aishell/ASR/transformer_ctc/train.py
+++ b/egs/aishell/ASR/transformer_ctc/train.py
@@ -691,6 +691,7 @@ def run(rank, world_size, args, wb=None):
             valid_dl=valid_dl,
             tb_writer=tb_writer,
             world_size=world_size,
+            wb=wb,
         )
 
         save_checkpoint(