Skip to content

Commit 7e65135

Browse files
committed
Fix ksponspeech preprocessing
1 parent 8a66128 commit 7e65135

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

openspeech/datasets/ksponspeech/lit_data_module.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ def prepare_data(self):
131131
raise FileNotFoundError
132132
self._generate_manifest_files(self.configs.dataset.manifest_file_path)
133133

134-
def setup(self, stage: Optional[str] = None, tokenizer: Tokenizer = None):
134+
def setup(self, stage: Optional[str] = None) -> None:
135135
r"""
136136
Split `train` and `valid` dataset for training.
137137
@@ -167,8 +167,6 @@ def setup(self, stage: Optional[str] = None, tokenizer: Tokenizer = None):
167167
dataset_path=dataset_path,
168168
audio_paths=audio_paths[stage],
169169
transcripts=transcripts[stage],
170-
sos_id=tokenizer.sos_id,
171-
eos_id=tokenizer.eos_id,
172170
apply_spec_augment=self.configs.audio.apply_spec_augment if stage == 'train' else False,
173171
del_silence=self.configs.audio.del_silence if stage == 'train' else False,
174172
)

0 commit comments

Comments
 (0)