mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 01:52:41 +00:00
update icefall/__init__.py to import more common functions. (#294)
This commit is contained in:
parent
cb3ba16f2b
commit
ceeb95bcb8
@ -1,3 +1,34 @@
|
|||||||
|
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 (
|
from .utils import (
|
||||||
AttributeDict,
|
AttributeDict,
|
||||||
MetricsTracker,
|
MetricsTracker,
|
||||||
|
@ -1,5 +1,6 @@
|
|||||||
[tool.isort]
|
[tool.isort]
|
||||||
profile = "black"
|
profile = "black"
|
||||||
|
skip = ["icefall/__init__.py"]
|
||||||
|
|
||||||
[tool.black]
|
[tool.black]
|
||||||
line-length = 80
|
line-length = 80
|
||||||
|
Loading…
x
Reference in New Issue
Block a user