fix comments

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

View File

@ -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:

View File

@ -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)

View File

@ -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