minor fixes

This commit is contained in:
JinZr 2023-08-19 21:44:16 +08:00
parent 15f6dcff9a
commit 9594efd782
2 changed files with 8 additions and 8 deletions

View File

@ -391,26 +391,26 @@ class SwitchBoardAsrDataModule:
@lru_cache() @lru_cache()
def train_all_cuts(self) -> CutSet: def train_all_cuts(self) -> CutSet:
logging.info("switchboard: About to get train cuts") logging.info("SwitchBoard: About to get train cuts")
return load_manifest_lazy( return load_manifest_lazy(
self.args.manifest_dir / "swbd_cuts_all.jsonl.gz" self.args.manifest_dir / "swbd_cuts_all.jsonl.gz"
).subset(last=2388) ).subset(last=166844)
@lru_cache() @lru_cache()
def dev_cuts(self) -> CutSet: def dev_cuts(self) -> CutSet:
logging.info("switchboard: About to get dev cuts") logging.info("SwitchBoard: About to get dev cuts")
return load_manifest_lazy( return load_manifest_lazy(
self.args.manifest_dir / "swbd_cuts_all.jsonl.gz" self.args.manifest_dir / "swbd_cuts_all.jsonl.gz"
).subset(first=50) ).subset(first=300)
@lru_cache() @lru_cache()
def test_eval2000_cuts(self) -> CutSet: def test_eval2000_cuts(self) -> CutSet:
logging.info("switchboard: About to get eval2000 cuts") logging.info("SwitchBoard: About to get eval2000 cuts")
return load_manifest_lazy( return load_manifest_lazy(
self.args.manifest_dir / "eval2000" / "eval2000_cuts_all.jsonl.gz" self.args.manifest_dir / "eval2000" / "eval2000_cuts_all.jsonl.gz"
) )
@lru_cache() @lru_cache()
def test_rt03_cuts(self) -> CutSet: def test_rt03_cuts(self) -> CutSet:
logging.info("switchboard: About to get rt03 cuts") logging.info("SwitchBoard: About to get rt03 cuts")
return load_manifest_lazy(self.args.manifest_dir / "swbd_cuts_rt03.jsonl.gz") return load_manifest_lazy(self.args.manifest_dir / "swbd_cuts_rt03.jsonl.gz")

View File

@ -142,7 +142,7 @@ if [ $stage -le 2 ] && [ $stop_stage -ge 2 ]; then
fi fi
if [ $stage -le 3 ] && [ $stop_stage -ge 3 ]; then if [ $stage -le 3 ] && [ $stop_stage -ge 3 ]; then
log "Stage 3: Compute fbank for SwitchBoard" log "Stage 3 I: Compute fbank for SwitchBoard"
if [ ! -e data/fbank/.swbd.done ]; then if [ ! -e data/fbank/.swbd.done ]; then
mkdir -p data/fbank/swbd_split${num_splits}/ mkdir -p data/fbank/swbd_split${num_splits}/
for index in $(seq 1 16); do for index in $(seq 1 16); do
@ -156,7 +156,7 @@ if [ $stage -le 3 ] && [ $stop_stage -ge 3 ]; then
fi fi
if [ $stage -le 3 ] && [ $stop_stage -ge 3 ]; then if [ $stage -le 3 ] && [ $stop_stage -ge 3 ]; then
log "Stage 3: Compute fbank for eval2000" log "Stage 3 II: Compute fbank for eval2000"
if [ ! -e data/fbank/.eval2000.done ]; then if [ ! -e data/fbank/.eval2000.done ]; then
mkdir -p data/fbank/eval2000/ mkdir -p data/fbank/eval2000/
./local/compute_fbank_eval2000.py ./local/compute_fbank_eval2000.py