|
1 |
| -import sys, os |
| 1 | +import os |
2 | 2 |
|
3 |
| -now_dir = os.getcwd() |
4 |
| -sys.path.append(os.path.join(now_dir)) |
5 |
| -sys.path.append(os.path.join(now_dir, "train")) |
6 |
| -import utils |
| 3 | +from lib.train import utils |
7 | 4 | import datetime
|
8 | 5 |
|
9 | 6 | hps = utils.get_hparams()
|
10 | 7 | os.environ["CUDA_VISIBLE_DEVICES"] = hps.gpus.replace("-", ",")
|
11 | 8 | n_gpus = len(hps.gpus.split("-"))
|
12 | 9 | from random import shuffle, randint
|
13 |
| -import traceback, json, argparse, itertools, math, torch, pdb |
14 | 10 |
|
| 11 | +import torch |
15 | 12 | torch.backends.cudnn.deterministic = False
|
16 | 13 | torch.backends.cudnn.benchmark = False
|
17 |
| -from torch import nn, optim |
18 | 14 | from torch.nn import functional as F
|
19 | 15 | from torch.utils.data import DataLoader
|
20 | 16 | from torch.utils.tensorboard import SummaryWriter
|
|
25 | 21 | from lib.infer_pack import commons
|
26 | 22 | from time import sleep
|
27 | 23 | from time import time as ttime
|
28 |
| -from data_utils import ( |
| 24 | +from lib.train.data_utils import ( |
29 | 25 | TextAudioLoaderMultiNSFsid,
|
30 | 26 | TextAudioLoader,
|
31 | 27 | TextAudioCollateMultiNSFsid,
|
|
45 | 41 | SynthesizerTrnMs768NSFsid_nono as RVC_Model_nof0,
|
46 | 42 | MultiPeriodDiscriminatorV2 as MultiPeriodDiscriminator,
|
47 | 43 | )
|
48 |
| -from losses import generator_loss, discriminator_loss, feature_loss, kl_loss |
49 |
| -from mel_processing import mel_spectrogram_torch, spec_to_mel_torch |
50 |
| -from process_ckpt import savee |
| 44 | +from lib.train.losses import generator_loss, discriminator_loss, feature_loss, kl_loss |
| 45 | +from lib.train.mel_processing import mel_spectrogram_torch, spec_to_mel_torch |
| 46 | +from lib.train.process_ckpt import savee |
51 | 47 |
|
52 | 48 | global_step = 0
|
53 | 49 |
|
|
0 commit comments