diff --git a/egs/aishell/ASR/transformer_ctc/.train.py.swp b/egs/aishell/ASR/transformer_ctc/.train.py.swp index 7b1881df5..6bd42da0f 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 adf6ffe62..7ed9e6e31 100755 --- a/egs/aishell/ASR/transformer_ctc/train.py +++ b/egs/aishell/ASR/transformer_ctc/train.py @@ -573,6 +573,7 @@ def train_one_epoch( graph_compiler=graph_compiler, valid_dl=valid_dl, world_size=world_size, + wb=wb, ) model.train() logging.info(f"Epoch {params.cur_epoch}, validation: {valid_info}")