Small fix to relative path

This commit is contained in:
Xinyuan Li 2024-01-09 12:41:05 -05:00
parent f1968bdf1f
commit bfd341d395

View File

@ -22,7 +22,7 @@ from typing import List, Tuple
import torch
import torch.nn as nn
from asr_datamodule import SluDataModule
from transducer.asr_datamodule import SluDataModule
from transducer.beam_search import greedy_search
from transducer.decoder import Decoder
from transducer.encoder import Tdnn
@ -336,7 +336,8 @@ def main():
model=model,
)
save_results(exp_dir=params.exp_dir, test_set_name="test_set", results=results)
test_set_name=str(args.feature_dir).split('/')[-2]
save_results(exp_dir=params.exp_dir, test_set_name=test_set_name, results=results)
logging.info("Done!")