提交 7ceef6c3 编写于 作者: H Hui Zhang

format code

上级 f9221b4b
......@@ -13,7 +13,7 @@ bc flac jq vim tig tree pkg-config libsndfile1 libflac-dev libvorbis-dev libboos
```
build-essential cmake libboost-system-dev libboost-thread-dev libboost-program-options-dev libboost-test-dev libeigen3-dev zlib1g-dev libbz2-dev liblzma-dev gcc-5 g++-5
```
### The dependencies of sox:
```
......@@ -25,7 +25,7 @@ libvorbis-dev libmp3lame-dev libmad-ocaml-dev
```
kenlm
sox
sox
mfa
openblas
kaldi
......
......@@ -554,10 +554,11 @@ class U2Tester(U2Trainer):
@paddle.no_grad()
def align(self):
ctc_utils.ctc_align(self.config,
self.model, self.align_loader, self.config.decoding.batch_size,
self.align_loader.collate_fn.stride_ms,
self.align_loader.collate_fn.vocab_list, self.args.result_file)
ctc_utils.ctc_align(self.config, self.model, self.align_loader,
self.config.decoding.batch_size,
self.align_loader.collate_fn.stride_ms,
self.align_loader.collate_fn.vocab_list,
self.args.result_file)
def load_inferspec(self):
"""infer model and input spec.
......
......@@ -527,10 +527,11 @@ class U2Tester(U2Trainer):
@paddle.no_grad()
def align(self):
ctc_utils.ctc_align(self.config,
self.model, self.align_loader, self.config.decoding.batch_size,
self.align_loader.collate_fn.stride_ms,
self.align_loader.collate_fn.vocab_list, self.args.result_file)
ctc_utils.ctc_align(self.config, self.model, self.align_loader,
self.config.decoding.batch_size,
self.align_loader.collate_fn.stride_ms,
self.align_loader.collate_fn.vocab_list,
self.args.result_file)
def load_inferspec(self):
"""infer model and input spec.
......
......@@ -543,10 +543,10 @@ class U2STTester(U2STTrainer):
@paddle.no_grad()
def align(self):
ctc_utils.ctc_align(self.config,
self.model, self.align_loader, self.config.decoding.batch_size,
self.config.collator.stride_ms,
self.vocab_list, self.args.result_file)
ctc_utils.ctc_align(self.config, self.model, self.align_loader,
self.config.decoding.batch_size,
self.config.collator.stride_ms, self.vocab_list,
self.args.result_file)
def load_inferspec(self):
"""infer model and input spec.
......
......@@ -12,8 +12,9 @@
# See the License for the specific language governing permissions and
# limitations under the License.
# Modified from wenet(https://github.com/wenet-e2e/wenet)
from typing import List
from pathlib import Path
from typing import List
import numpy as np
import paddle
......
......@@ -67,19 +67,16 @@ class LJSpeechCollector(object):
# Sort by text_len in descending order
texts = [
i
for i, _ in sorted(
i for i, _ in sorted(
zip(texts, text_lens), key=lambda x: x[1], reverse=True)
]
mels = [
i
for i, _ in sorted(
i for i, _ in sorted(
zip(mels, text_lens), key=lambda x: x[1], reverse=True)
]
mel_lens = [
i
for i, _ in sorted(
i for i, _ in sorted(
zip(mel_lens, text_lens), key=lambda x: x[1], reverse=True)
]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册