From 54bcc167e17e0edb26f9d32a1f35510fbf323fee Mon Sep 17 00:00:00 2001 From: Guo Liyong Date: Thu, 2 Dec 2021 17:46:14 +0800 Subject: [PATCH] fix ci --- egs/librispeech/ASR/conformer_ctc/decode.py | 9 ++++++--- egs/librispeech/ASR/conformer_ctc/train.py | 2 +- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/egs/librispeech/ASR/conformer_ctc/decode.py b/egs/librispeech/ASR/conformer_ctc/decode.py index f353adf37..ed2da7b76 100755 --- a/egs/librispeech/ASR/conformer_ctc/decode.py +++ b/egs/librispeech/ASR/conformer_ctc/decode.py @@ -501,7 +501,8 @@ def save_results( for key, results in results_dict.items(): recog_path = ( params.exp_dir - / f"epoch-{params.epoch}-avg-{params.avg}-recogs-{test_set_name}-{key}.txt" + / f"epoch-{params.epoch}-avg-{params.avg}- \ + recogs-{test_set_name}-{key}.txt" ) store_transcripts(filename=recog_path, texts=results) if enable_log: @@ -511,7 +512,8 @@ def save_results( # ref/hyp pairs. errs_filename = ( params.exp_dir - / f"epoch-{params.epoch}-avg-{params.avg}-errs-{test_set_name}-{key}.txt" + / f"epoch-{params.epoch}-avg-{params.avg}- \ + errs-{test_set_name}-{key}.txt" ) with open(errs_filename, "w") as f: wer = write_error_stats( @@ -527,7 +529,8 @@ def save_results( test_set_wers = sorted(test_set_wers.items(), key=lambda x: x[1]) errs_info = ( params.exp_dir - / f"epoch-{params.epoch}-avg-{params.avg}-wer-summary-{test_set_name}.txt" + / f"epoch-{params.epoch}-avg-{params.avg}- \ + wer-summary-{test_set_name}.txt" ) with open(errs_info, "w") as f: print("settings\tWER", file=f) diff --git a/egs/librispeech/ASR/conformer_ctc/train.py b/egs/librispeech/ASR/conformer_ctc/train.py index ceafe6cbe..8a9bcfa8b 100755 --- a/egs/librispeech/ASR/conformer_ctc/train.py +++ b/egs/librispeech/ASR/conformer_ctc/train.py @@ -439,7 +439,7 @@ def compute_loss( info["att_loss"] = att_loss.detach().cpu().item() if params.codebook_weight != 0.0: - info["codebook_loss"] = cdidx_loss.detach().cpu().item() + info["codebook_loss"] = codebook_loss.detach().cpu().item() info["loss"] = loss.detach().cpu().item()