From 2c5902d7c58d9ec437637332015a836ee05f6db7 Mon Sep 17 00:00:00 2001 From: huangyuxin Date: Wed, 29 Dec 2021 03:13:14 +0000 Subject: [PATCH] rename decoding to decode --- paddlespeech/s2t/exps/u2/config.py | 1 + paddlespeech/s2t/exps/u2/model.py | 8 ++++---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/paddlespeech/s2t/exps/u2/config.py b/paddlespeech/s2t/exps/u2/config.py index 59376e95..537d0c21 100644 --- a/paddlespeech/s2t/exps/u2/config.py +++ b/paddlespeech/s2t/exps/u2/config.py @@ -31,6 +31,7 @@ U2Trainer.params(_C) _C.decoding = U2Tester.params() + def get_cfg_defaults(): """Get a yacs CfgNode object with default values for my_project.""" # Return a clone so that the defaults will not be altered diff --git a/paddlespeech/s2t/exps/u2/model.py b/paddlespeech/s2t/exps/u2/model.py index 1de9541d..31610e15 100644 --- a/paddlespeech/s2t/exps/u2/model.py +++ b/paddlespeech/s2t/exps/u2/model.py @@ -294,7 +294,7 @@ class U2Trainer(Trainer): json_file=config.test_manifest, train_mode=False, sortagrad=False, - batch_size=config.decoding.decode_batch_size, + batch_size=config.decode.decode_batch_size, maxlen_in=float('inf'), maxlen_out=float('inf'), minibatches=0, @@ -313,7 +313,7 @@ class U2Trainer(Trainer): json_file=config.test_manifest, train_mode=False, sortagrad=False, - batch_size=config.decoding.decode_batch_size, + batch_size=config.decode.decode_batch_size, maxlen_in=float('inf'), maxlen_out=float('inf'), minibatches=0, @@ -452,7 +452,7 @@ class U2Tester(U2Trainer): texts, texts_len, fout=None): - decode_config = self.config.decoding + decode_config = self.config.decode errors_sum, len_refs, num_ins = 0.0, 0, 0 errors_func = error_rate.char_errors if decode_config.error_rate_type == 'cer' else error_rate.word_errors error_rate_func = error_rate.cer if decode_config.error_rate_type == 'cer' else error_rate.wer @@ -564,7 +564,7 @@ class U2Tester(U2Trainer): @paddle.no_grad() def align(self): ctc_utils.ctc_align(self.config, self.model, self.align_loader, - self.config.decoding.decode_batch_size, + self.config.decode.decode_batch_size, self.config.stride_ms, self.vocab_list, self.args.result_file)