mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-11 02:52:18 +00:00
6e609c67a2
* support streaming in conformer * Add more documents * support streaming on pruned_transducer_stateless2; add delay penalty; fixes for decode states * Minor fixes * streaming for pruned_transducer_stateless4 * Fix conv cache error, support async streaming decoding * Fix style * Fix style * Fix style * Add torch.jit.export * mask the initial cache * Cutting off invalid frames of encoder_embed output * fix relative positional encoding in streaming decoding for compution saving * Minor fixes * Minor fixes * Minor fixes * Minor fixes * Minor fixes * Fix jit export for torch 1.6 * Minor fixes for streaming decoding * Minor fixes on decode stream * move model parameters to train.py * make states in forward streaming optional * update pretrain to support streaming model * update results.md * update tensorboard and pre-models * fix typo * Fix tests * remove unused arguments * add streaming decoding ci * Minor fix * Minor fix * disable right context by default
67 lines
1.1 KiB
Python
67 lines
1.1 KiB
Python
# isort:skip_file
|
|
|
|
from . import (
|
|
checkpoint,
|
|
decode,
|
|
dist,
|
|
env,
|
|
utils
|
|
)
|
|
|
|
from .checkpoint import (
|
|
average_checkpoints,
|
|
find_checkpoints,
|
|
load_checkpoint,
|
|
remove_checkpoints,
|
|
save_checkpoint,
|
|
save_checkpoint_with_global_batch_idx,
|
|
)
|
|
|
|
from .decode import (
|
|
get_lattice,
|
|
nbest_decoding,
|
|
nbest_oracle,
|
|
one_best_decoding,
|
|
rescore_with_attention_decoder,
|
|
rescore_with_n_best_list,
|
|
rescore_with_whole_lattice,
|
|
)
|
|
|
|
from .dist import (
|
|
cleanup_dist,
|
|
setup_dist,
|
|
)
|
|
|
|
from .env import (
|
|
get_env_info,
|
|
get_git_branch_name,
|
|
get_git_date,
|
|
get_git_sha1,
|
|
)
|
|
|
|
from .utils import (
|
|
AttributeDict,
|
|
MetricsTracker,
|
|
add_eos,
|
|
add_sos,
|
|
concat,
|
|
encode_supervisions,
|
|
get_alignments,
|
|
get_executor,
|
|
get_texts,
|
|
l1_norm,
|
|
l2_norm,
|
|
linf_norm,
|
|
load_alignments,
|
|
make_pad_mask,
|
|
measure_gradient_norms,
|
|
measure_weight_norms,
|
|
optim_step_and_measure_param_change,
|
|
save_alignments,
|
|
setup_logger,
|
|
store_transcripts,
|
|
str2bool,
|
|
subsequent_chunk_mask,
|
|
write_error_stats,
|
|
)
|