21 lines
541 B
Python
21 lines
541 B
Python
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()
|