Skip to content

Commit f17a881

Browse files
author
luca.gaudino
committed
improve trafo lm opts
1 parent ef4a104 commit f17a881

File tree

7 files changed

+234
-334
lines changed

7 files changed

+234
-334
lines changed

users/gaudino/dump.py

Lines changed: 0 additions & 237 deletions
This file was deleted.

users/gaudino/experiments/rf_conformer_att_2023/librispeech_960/conformer_import_moh_att_2023_06_30.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -819,7 +819,7 @@ def from_scratch_model_def(
819819
from_scratch_model_def.batch_size_factor = (
820820
40 # change batch size here - 20 for att_window - 40 for ctc_prefix
821821
)
822-
from_scratch_model_def.max_seqs = 200 #1
822+
from_scratch_model_def.max_seqs = 1 # 200 # 1
823823

824824

825825
def from_scratch_training(

users/gaudino/experiments/rf_conformer_att_2023/librispeech_960/model_recogs/model_recog.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,6 @@ def model_recog(
168168
if model.search_args.get("add_trafo_lm", False):
169169
trafo_lm_out = model.trafo_lm(target, state=trafo_lm_state, spatial_dim=single_step_dim)
170170
trafo_lm_state = trafo_lm_out["state"]
171-
breakpoint()
172171
trafo_log_prob = rf.log_softmax(trafo_lm_out["output"], axis=model.target_dim)
173172
if i > 0:
174173
label_log_prob = (

0 commit comments

Comments
 (0)