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

remove bilingual tag from train.py

Browse Source
This commit is contained in:
Bailey Hirota 2025-05-14 08:37:44 +09:00
parent f210002500
commit ee2a6d60e0
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/multi_ja_en/ASR/zipformer/train.py
View File

@ -791,7 +791,7 @@ def compute_loss(
warm_step = params.warm_step warm_step = params.warm_step
texts = batch["supervisions"]["text"] texts = batch["supervisions"]["text"]
y = sentencepiece_processor.encode(texts, out_type=int) y = sentencepiece_processor.encode(texts, out_type=int)
y = k2.RaggedTensor(y) y = k2.RaggedTensor(y)
with torch.set_grad_enabled(is_training): with torch.set_grad_enabled(is_training):
@ -1178,7 +1178,7 @@ def run(rank, world_size, args):
if params.inf_check: if params.inf_check:
register_inf_check_hooks(model) register_inf_check_hooks(model)
multidataset_datamodule = MultiDatasetAsrDataModule(args) reazonspeech_corpus = ReazonSpeechAsrDataModule(args)
multi_dataset = MultiDataset(args) multi_dataset = MultiDataset(args)
@ -1244,7 +1244,7 @@ def run(rank, world_size, args):
) )
valid_cuts = multi_dataset.dev_cuts() valid_cuts = multi_dataset.dev_cuts()
valid_dl = multidataset_datamodule.valid_dataloaders(valid_cuts) valid_dl = reazonspeech_corpus.valid_dataloaders(valid_cuts)
if not params.print_diagnostics: if not params.print_diagnostics:
scan_pessimistic_batches_for_oom( scan_pessimistic_batches_for_oom(
@ -1386,7 +1386,7 @@ def scan_pessimistic_batches_for_oom(
def main(): def main():
parser = get_parser() parser = get_parser()
MultiDatasetAsrDataModule.add_arguments(parser) ReazonSpeechAsrDataModule.add_arguments(parser)
args = parser.parse_args() args = parser.parse_args()
args.exp_dir = Path(args.exp_dir) args.exp_dir = Path(args.exp_dir)