mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-10 18:42:19 +00:00
5c17255eec
* Sort result to make it more convenient to compare decoding results * Add cut_id to recognition results * add cut_id to results for all recipes * Fix torch.jit.script * Fix comments * Minor fixes * Fix torch.jit.tracing for Pytorch version before v1.9.0
68 lines
1.1 KiB
Python
68 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,
|
|
is_jit_tracing,
|
|
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,
|
|
)
|