diff --git a/egs/himia/wuw/ctc_tdnn/decode.py b/egs/himia/wuw/ctc_tdnn/decode.py index 9d05a3310..747f7ff52 100755 --- a/egs/himia/wuw/ctc_tdnn/decode.py +++ b/egs/himia/wuw/ctc_tdnn/decode.py @@ -83,7 +83,7 @@ class FiniteStateTransducer: if len(arc) == 4: # Non-final state assert max_state_id <= src_state_id, ( 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: new_state = State() @@ -95,7 +95,7 @@ class FiniteStateTransducer: else: assert ( 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 def to_str(self) -> None: diff --git a/egs/himia/wuw/ctc_tdnn/train.py b/egs/himia/wuw/ctc_tdnn/train.py index 04953d9c3..d35744e20 100755 --- a/egs/himia/wuw/ctc_tdnn/train.py +++ b/egs/himia/wuw/ctc_tdnn/train.py @@ -529,7 +529,7 @@ def run(rank, world_size, args): 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) diff --git a/egs/himia/wuw/run_ctc_tdnn.sh b/egs/himia/wuw/run_ctc_tdnn.sh index 258c2b2b1..2f6a1788f 100644 --- a/egs/himia/wuw/run_ctc_tdnn.sh +++ b/egs/himia/wuw/run_ctc_tdnn.sh @@ -26,7 +26,7 @@ if [ $stage -le 0 ] && [ $stop_stage -ge 0 ]; then log "Stage 0: Model training" python ./ctc_tdnn/train.py \ --num-epochs $epoch \ - --exp_dir $exp_dir + --exp-dir $exp_dir --max-duration $max_duration fi