You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
MockingBird/train.py

23 lines
587 B
Python

import sys
import torch
import argparse
import numpy as np
from utils.load_yaml import HpsYaml
from models.ppg2mel.train.train_linglf02mel_seq2seq_oneshotvc import Solver
def main():
# Arguments
2 years ago
preparser = argparse.ArgumentParser(description=
'Training model.')
preparser.add_argument('--type', type=str,
help='type of training ')
###
2 years ago
paras, _ = preparser.parse_known_args()
if paras.type == "synth":
from control.cli.synthesizer_train import new_train
new_train()
if __name__ == "__main__":
main()