mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-09-05 15:14:18 +00:00
Small fix to relative path
This commit is contained in:
parent
f1968bdf1f
commit
bfd341d395
@ -22,7 +22,7 @@ from typing import List, Tuple
|
|||||||
|
|
||||||
import torch
|
import torch
|
||||||
import torch.nn as nn
|
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.beam_search import greedy_search
|
||||||
from transducer.decoder import Decoder
|
from transducer.decoder import Decoder
|
||||||
from transducer.encoder import Tdnn
|
from transducer.encoder import Tdnn
|
||||||
@ -336,7 +336,8 @@ def main():
|
|||||||
model=model,
|
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!")
|
logging.info("Done!")
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user