mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 18:24:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix code style

Browse Source
This commit is contained in:
pkufool 2021-11-17 19:09:36 +08:00
parent 99b39bccce
commit f7a26400ab
2 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
egs/aishell/ASR/tdnn_lstm_ctc/decode.py
View File

@ -20,7 +20,7 @@ import argparse
import logging import logging
from collections import defaultdict from collections import defaultdict
from pathlib import Path from pathlib import Path
from typing import Dict, List, Optional, Tuple from typing import Dict, List, Tuple
import k2 import k2
import torch import torch
@ -29,12 +29,7 @@ from asr_datamodule import AishellAsrDataModule
from model import TdnnLstm from model import TdnnLstm
from icefall.checkpoint import average_checkpoints, load_checkpoint from icefall.checkpoint import average_checkpoints, load_checkpoint
from icefall.decode import ( from icefall.decode import get_lattice, nbest_decoding, one_best_decoding
get_lattice,
nbest_decoding,
one_best_decoding,
rescore_with_attention_decoder,
)
from icefall.lexicon import Lexicon from icefall.lexicon import Lexicon
from icefall.utils import ( from icefall.utils import (
AttributeDict, AttributeDict,

1
icefall/decode.py
View File

@ -903,4 +903,3 @@ def rescore_with_attention_decoder(
key = f"ngram_lm_scale_{n_scale}_attention_scale_{a_scale}" key = f"ngram_lm_scale_{n_scale}_attention_scale_{a_scale}"
ans[key] = best_path ans[key] = best_path
return ans return ans