Merge pull request #78 from sooftware/audio-conf

Update audio-conf

Author: sooftware

bin/main.py

@@ -30,7 +30,10 @@
     LibriSpeechVocabulary,
 )
 from kospeech.dataclasses import (
-    AudioConfig,
+    FBankConfig,
+    MelSpectrogramConfig,
+    MFCCConfig,
+    SpectrogramConfig,
     TrainConfig,
     DeepSpeech2Config,
     JointCTCAttentionLASConfig,
@@ -119,10 +122,10 @@ def train(config: DictConfig):
 
 
 cs = ConfigStore.instance()
-cs.store(group="audio", name="fbank", node=AudioConfig, package="audio")
-cs.store(group="audio", name="melspectrogram", node=AudioConfig, package="audio")
-cs.store(group="audio", name="mfcc", node=AudioConfig, package="audio")
-cs.store(group="audio", name="spectrogram", node=AudioConfig, package="audio")
+cs.store(group="audio", name="fbank", node=FBankConfig, package="audio")
+cs.store(group="audio", name="melspectrogram", node=MelSpectrogramConfig, package="audio")
+cs.store(group="audio", name="mfcc", node=MFCCConfig, package="audio")
+cs.store(group="audio", name="spectrogram", node=SpectrogramConfig, package="audio")
 cs.store(group="train", name="ds2_train", node=TrainConfig, package="train")
 cs.store(group="train", name="las_train", node=TrainConfig, package="train")
 cs.store(group="train", name="transformer_train", node=TrainConfig, package="train")