mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 18:24:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor changes

Browse Source
This commit is contained in:
marcoyang 2024-03-26 15:05:35 +08:00
parent f4c187286a
commit 64dbcd07c5
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
egs/audioset/AT/zipformer/at_datamodule.py
View File

@ -91,7 +91,7 @@ class AudioSetATDatamodule:
"--manifest-dir", "--manifest-dir",
type=Path, type=Path,
default=Path("data/fbank"), default=Path("data/fbank"),
help="Path to directory with train/valid/test cuts.", help="Path to directory with audioset train/test cuts.",
) )
group.add_argument( group.add_argument(
"--max-duration", "--max-duration",
@ -395,7 +395,7 @@ class AudioSetATDatamodule:
@lru_cache() @lru_cache()
def audioset_train_cuts(self) -> CutSet: def audioset_train_cuts(self) -> CutSet:
logging.info("About to get the audioset cuts for KD.") logging.info("About to get the audioset training cuts.")
balanced_cuts = load_manifest_lazy( balanced_cuts = load_manifest_lazy(
self.args.manifest_dir / "cuts_audioset_balanced.jsonl.gz" self.args.manifest_dir / "cuts_audioset_balanced.jsonl.gz"
) )
@ -415,7 +415,7 @@ class AudioSetATDatamodule:
@lru_cache() @lru_cache()
def audioset_eval_cuts(self) -> CutSet: def audioset_eval_cuts(self) -> CutSet:
logging.info("About to get test-other cuts") logging.info("About to get audioset eval cuts")
return load_manifest_lazy( return load_manifest_lazy(
self.args.manifest_dir / "cuts_audioset_eval.jsonl.gz" self.args.manifest_dir / "cuts_audioset_eval.jsonl.gz"
) )