mocking-bird / train.py
smf2010's picture
Upload 204 files
a23d717 verified
import argparse
def main():
# Arguments
preparser = argparse.ArgumentParser(description=
'Training model.')
preparser.add_argument('--type', type=str,
help='type of training ')
###
paras, _ = preparser.parse_known_args()
if paras.type == "synth":
from control.cli.synthesizer_train import new_train
new_train()
if paras.type == "vits":
from models.synthesizer.train_vits import new_train
new_train()
if __name__ == "__main__":
main()