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

small fixes

Browse Source
This commit is contained in:
k2-fsa 2025-07-01 13:41:56 +08:00
parent b4e9edbed1
commit 127b4985d3
5 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/conformer_ctc3/train.py
View File

@ -92,13 +92,13 @@ from icefall.env import get_env_info
from icefall.graph_compiler import CtcTrainingGraphCompiler from icefall.graph_compiler import CtcTrainingGraphCompiler
from icefall.lexicon import Lexicon from icefall.lexicon import Lexicon
from icefall.utils import ( from icefall.utils import (
create_grad_scaler,
torch_autocast,
AttributeDict, AttributeDict,
MetricsTracker, MetricsTracker,
create_grad_scaler,
encode_supervisions, encode_supervisions,
setup_logger, setup_logger,
str2bool, str2bool,
torch_autocast,
) )
LRSchedulerType = Union[torch.optim.lr_scheduler._LRScheduler, optim.LRScheduler] LRSchedulerType = Union[torch.optim.lr_scheduler._LRScheduler, optim.LRScheduler]

1
egs/librispeech/ASR/pruned_transducer_stateless7/finetune.py
View File

@ -84,6 +84,7 @@ from icefall.hooks import register_inf_check_hooks
from icefall.utils import ( from icefall.utils import (
AttributeDict, AttributeDict,
MetricsTracker, MetricsTracker,
create_grad_scaler,
filter_uneven_sized_batch, filter_uneven_sized_batch,
setup_logger, setup_logger,
str2bool, str2bool,

1
egs/librispeech/ASR/pruned_transducer_stateless7_ctc_bs/train.py
View File

@ -81,6 +81,7 @@ from icefall.hooks import register_inf_check_hooks
from icefall.utils import ( from icefall.utils import (
AttributeDict, AttributeDict,
MetricsTracker, MetricsTracker,
create_grad_scaler,
encode_supervisions, encode_supervisions,
setup_logger, setup_logger,
str2bool, str2bool,

2
egs/librispeech/ASR/zipformer_mmi/train.py
View File

@ -86,11 +86,11 @@ from icefall.mmi_graph_compiler import MmiTrainingGraphCompiler
from icefall.utils import ( from icefall.utils import (
AttributeDict, AttributeDict,
MetricsTracker, MetricsTracker,
create_grad_scaler,
encode_supervisions, encode_supervisions,
setup_logger, setup_logger,
str2bool, str2bool,
torch_autocast, torch_autocast,
create_grad_scaler,
) )
LRSchedulerType = Union[torch.optim.lr_scheduler._LRScheduler, optim.LRScheduler] LRSchedulerType = Union[torch.optim.lr_scheduler._LRScheduler, optim.LRScheduler]

2
egs/librispeech/SSL/hubert/pretrain.py
View File

@ -75,12 +75,12 @@ from icefall.dist import cleanup_dist, setup_dist
from icefall.env import get_env_info from icefall.env import get_env_info
from icefall.hooks import register_inf_check_hooks from icefall.hooks import register_inf_check_hooks
from icefall.utils import ( from icefall.utils import (
torch_autocast,
AttributeDict, AttributeDict,
MetricsTracker, MetricsTracker,
get_parameter_groups_with_lrs, get_parameter_groups_with_lrs,
setup_logger, setup_logger,
str2bool, str2bool,
torch_autocast,
) )
LRSchedulerType = Union[torch.optim.lr_scheduler._LRScheduler, optim.LRScheduler] LRSchedulerType = Union[torch.optim.lr_scheduler._LRScheduler, optim.LRScheduler]