minor updates

This commit is contained in:
JinZr 2023-09-28 10:53:31 +08:00
parent 09ada8fb48
commit 8f2d480439
2 changed files with 2 additions and 2 deletions

View File

@ -32,7 +32,7 @@ class MultiDataset:
It is expected to contain the following files:
- aishell2_cuts_train.jsonl.gz
"""
self.fbank_dir = Path(args.fbank_dir)
self.fbank_dir = Path(args.manifest_dir)
self.use_tal_csasr = args.use_tal_csasr
def train_cuts(self) -> CutSet:

View File

@ -1180,7 +1180,7 @@ def run(rank, world_size, args):
register_inf_check_hooks(model)
data_module = AsrDataModule(args)
multi_dataset = MultiDataset(args.manifest_dir)
multi_dataset = MultiDataset(args)
train_cuts = multi_dataset.train_cuts()