from local

This commit is contained in:
dohe0342 2023-06-09 17:29:32 +09:00
parent ec5c1447ab
commit 9019981309
3 changed files with 2 additions and 2 deletions

Binary file not shown.

View File

@ -48,7 +48,7 @@ import k2
import sentencepiece as spm
import torch
import torch.nn as nn
from asr_datamodule import LibriSpeechAsrDataModule
from asr_datamodule import TedLiumAsrDataModule
from beam_search import (
beam_search,
fast_beam_search_nbest,
@ -629,7 +629,7 @@ def save_results(
@torch.no_grad()
def main():
parser = get_parser()
LibriSpeechAsrDataModule.add_arguments(parser)
TedLiumAsrDataModule.add_arguments(parser)
args = parser.parse_args()
args.exp_dir = Path(args.exp_dir)