Fix style checking

This commit is contained in:
pkufool 2021-08-24 21:08:11 +08:00
parent 88bb4046af
commit fb98c5d1bf
3 changed files with 3 additions and 1 deletions

View File

@ -5,6 +5,7 @@ max-line-length = 80
per-file-ignores = per-file-ignores =
# line too long # line too long
egs/librispeech/ASR/conformer_ctc/conformer.py: E501, egs/librispeech/ASR/conformer_ctc/conformer.py: E501,
egs/librispeech/ASR/conformer_ctc/decode.py: E501,
exclude = exclude =
.git, .git,

View File

@ -43,6 +43,7 @@ from icefall.utils import (
setup_logger, setup_logger,
store_transcripts, store_transcripts,
write_error_stats, write_error_stats,
str2bool,
) )

View File

@ -219,7 +219,7 @@ def main():
features, batch_first=True, padding_value=math.log(1e-10) features, batch_first=True, padding_value=math.log(1e-10)
) )
features = features.permute(0, 2, 1) # now features is [N, C, T] features = features.permute(0, 2, 1) # now features is [N, C, T]
with torch.no_grad(): with torch.no_grad():
nnet_output = model(features) nnet_output = model(features)
# nnet_output is [N, T, C] # nnet_output is [N, T, C]