mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-07 16:14:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor updates

Browse Source
This commit is contained in:
marcoyang 2023-09-27 11:31:14 +08:00
parent ae3149cb7f
commit e058ba0a65
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/libriheavy/ASR/zipformer_prompt_asr/decode_bert.py
View File

@ -935,7 +935,8 @@ def main():
filename_start=filename_start, filename_start=filename_start,
filename_end=filename_end, filename_end=filename_end,
device=device, device=device,
) ),
strict=False,
) )
model.to(device) model.to(device)

11
egs/libriheavy/ASR/zipformer_prompt_asr/train_bert_encoder.py
View File

@ -334,6 +334,12 @@ def add_model_arguments(parser: argparse.ArgumentParser):
"chunk left-context frames will be chosen randomly from this list; else not relevant.", "chunk left-context frames will be chosen randomly from this list; else not relevant.",
) )
parser.add_argument(
"--freeze-text-encoder",
type=str2bool,
default=True,
)
parser.add_argument( parser.add_argument(
"--text-encoder-type", "--text-encoder-type",
type=str, type=str,
@ -586,11 +592,6 @@ def get_parser():
default=0.05, default=0.05,
help="The probability of masking prompts", help="The probability of masking prompts",
) )
parser.add_argument(
"--freeze-text-encoder",
type=str2bool,
default=True,
)
parser.add_argument( parser.add_argument(
"--forced-upper-pre-text", "--forced-upper-pre-text",