import sys import torch import argparse import numpy as np from utils.hparams import HpsYaml from models.ppg2mel.train.train_linglf02mel_seq2seq_oneshotvc import Solver # For reproducibility, comment these may speed up training torch.backends.cudnn.deterministic = True torch.backends.cudnn.benchmark = False def main(): # Arguments parser = argparse.ArgumentParser(description= 'Training PPG2Mel VC model.') parser.add_argument('--config', type=str, help='Path to experiment config, e.g., config/vc.yaml') parser.add_argument('--name', default=None, type=str, help='Name for logging.') parser.add_argument('--logdir', default='log/', type=str, help='Logging path.', required=False) parser.add_argument('--ckpdir', default='ppg2mel/saved_models/', type=str, help='Checkpoint path.', required=False) parser.add_argument('--outdir', default='result/', type=str, help='Decode output path.', required=False) parser.add_argument('--load', default=None, type=str, help='Load pre-trained model (for training only)', required=False) parser.add_argument('--warm_start', action='store_true', help='Load model weights only, ignore specified layers.') parser.add_argument('--seed', default=0, type=int, help='Random seed for reproducable results.', required=False) parser.add_argument('--njobs', default=8, type=int, help='Number of threads for dataloader/decoding.', required=False) parser.add_argument('--cpu', action='store_true', help='Disable GPU training.') parser.add_argument('--no-pin', action='store_true', help='Disable pin-memory for dataloader') parser.add_argument('--test', action='store_true', help='Test the model.') parser.add_argument('--no-msg', action='store_true', help='Hide all messages.') parser.add_argument('--finetune', action='store_true', help='Finetune model') parser.add_argument('--oneshotvc', action='store_true', help='Oneshot VC model') parser.add_argument('--bilstm', action='store_true', help='BiLSTM VC model') parser.add_argument('--lsa', action='store_true', help='Use location-sensitive attention (LSA)') ### paras = parser.parse_args() setattr(paras, 'gpu', not paras.cpu) setattr(paras, 'pin_memory', not paras.no_pin) setattr(paras, 'verbose', not paras.no_msg) # Make the config dict dot visitable config = HpsYaml(paras.config) np.random.seed(paras.seed) torch.manual_seed(paras.seed) if torch.cuda.is_available(): torch.cuda.manual_seed_all(paras.seed) print(">>> OneShot VC training ...") mode = "train" solver = Solver(config, paras, mode) solver.load_data() solver.set_model() solver.exec() print(">>> Oneshot VC train finished!") sys.exit(0) if __name__ == "__main__": main()