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

fix comments

Browse Source
This commit is contained in:
glynpu 2023-03-16 20:17:44 +08:00
parent fe6dcd7429
commit b769f10bf6
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/himia/wuw/ctc_tdnn/decode.py
View File

@ -83,7 +83,7 @@ class FiniteStateTransducer:
if len(arc) == 4: # Non-final state if len(arc) == 4: # Non-final state
assert max_state_id <= src_state_id, ( assert max_state_id <= src_state_id, (
f"Fsa must be sorted by src_state, " f"Fsa must be sorted by src_state, "
f"while {cur_number_states} <= {src_state_id}. Check your graph." f"while {max_state_id} <= {src_state_id}. Check your graph."
) )
if max_state_id < src_state_id: if max_state_id < src_state_id:
new_state = State() new_state = State()
@ -95,7 +95,7 @@ class FiniteStateTransducer:
else: else:
assert ( assert (
max_state_id == src_state_id max_state_id == src_state_id
), f"Final state seems unreachable. Check your graph." ), "Final state seems unreachable. Check your graph."
self.final_state_id = src_state_id self.final_state_id = src_state_id
def to_str(self) -> None: def to_str(self) -> None:

2
egs/himia/wuw/ctc_tdnn/train.py
View File

@ -529,7 +529,7 @@ def run(rank, world_size, args):
logging.info("About to create model") logging.info("About to create model")
model = Tdnn(params.feature_dim, params.numb_class) model = Tdnn(params.feature_dim, params.num_class)
checkpoints = load_checkpoint_if_available(params=params, model=model) checkpoints = load_checkpoint_if_available(params=params, model=model)

2
egs/himia/wuw/run_ctc_tdnn.sh
View File

@ -26,7 +26,7 @@ if [ $stage -le 0 ] && [ $stop_stage -ge 0 ]; then
log "Stage 0: Model training" log "Stage 0: Model training"
python ./ctc_tdnn/train.py \ python ./ctc_tdnn/train.py \
--num-epochs $epoch \ --num-epochs $epoch \
--exp_dir $exp_dir --exp-dir $exp_dir
--max-duration $max_duration --max-duration $max_duration
fi fi