Skip to content

Commit f70da25

Browse files
committed
fix: code lint by optimizing train lib's importing
1 parent add253b commit f70da25

File tree

7 files changed

+7
-11
lines changed

7 files changed

+7
-11
lines changed
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

train_nsf_sim_cache_sid_load_pretrain.py

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,16 @@
1-
import sys, os
1+
import os
22

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
74
import datetime
85

96
hps = utils.get_hparams()
107
os.environ["CUDA_VISIBLE_DEVICES"] = hps.gpus.replace("-", ",")
118
n_gpus = len(hps.gpus.split("-"))
129
from random import shuffle, randint
13-
import traceback, json, argparse, itertools, math, torch, pdb
1410

11+
import torch
1512
torch.backends.cudnn.deterministic = False
1613
torch.backends.cudnn.benchmark = False
17-
from torch import nn, optim
1814
from torch.nn import functional as F
1915
from torch.utils.data import DataLoader
2016
from torch.utils.tensorboard import SummaryWriter
@@ -25,7 +21,7 @@
2521
from lib.infer_pack import commons
2622
from time import sleep
2723
from time import time as ttime
28-
from data_utils import (
24+
from lib.train.data_utils import (
2925
TextAudioLoaderMultiNSFsid,
3026
TextAudioLoader,
3127
TextAudioCollateMultiNSFsid,
@@ -45,9 +41,9 @@
4541
SynthesizerTrnMs768NSFsid_nono as RVC_Model_nof0,
4642
MultiPeriodDiscriminatorV2 as MultiPeriodDiscriminator,
4743
)
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
5147

5248
global_step = 0
5349

0 commit comments

Comments
 (0)