mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-18 07:41:03 +00:00
ZDisket
456fc04e63
- Tacotron2 based on https://github.com/keonlee9420/Comprehensive-Tacotron2 - Removed DeepSpeaker and Tensorflow dependency - Added ARPA function - Replaced guided attention loss with Convolutional Attention Consistency
28 lines
640 B
Python
28 lines
640 B
Python
import argparse
|
|
|
|
from utils.tools import get_configs_of
|
|
from preprocessor import ljspeech, vctk
|
|
|
|
|
|
def main(config):
|
|
if "LJSpeech" in config["dataset"]:
|
|
preprocessor = ljspeech.Preprocessor(config)
|
|
if "VCTK" in config["dataset"]:
|
|
preprocessor = vctk.Preprocessor(config)
|
|
|
|
preprocessor.build_from_path()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--dataset",
|
|
type=str,
|
|
required=True,
|
|
help="name of dataset",
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
preprocess_config, *_ = get_configs_of(args.dataset)
|
|
main(preprocess_config)
|