fix: code lint by optimizing train lib's importing

This commit is contained in:
源文雨 2023-07-23 12:08:11 +08:00
parent add253b476
commit f70da25f00
7 changed files with 7 additions and 11 deletions

View File

@ -1,20 +1,16 @@
import sys, os import os
now_dir = os.getcwd() from lib.train import utils
sys.path.append(os.path.join(now_dir))
sys.path.append(os.path.join(now_dir, "train"))
import utils
import datetime import datetime
hps = utils.get_hparams() hps = utils.get_hparams()
os.environ["CUDA_VISIBLE_DEVICES"] = hps.gpus.replace("-", ",") os.environ["CUDA_VISIBLE_DEVICES"] = hps.gpus.replace("-", ",")
n_gpus = len(hps.gpus.split("-")) n_gpus = len(hps.gpus.split("-"))
from random import shuffle, randint from random import shuffle, randint
import traceback, json, argparse, itertools, math, torch, pdb
import torch
torch.backends.cudnn.deterministic = False torch.backends.cudnn.deterministic = False
torch.backends.cudnn.benchmark = False torch.backends.cudnn.benchmark = False
from torch import nn, optim
from torch.nn import functional as F from torch.nn import functional as F
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter from torch.utils.tensorboard import SummaryWriter
@ -25,7 +21,7 @@ from torch.cuda.amp import autocast, GradScaler
from lib.infer_pack import commons from lib.infer_pack import commons
from time import sleep from time import sleep
from time import time as ttime from time import time as ttime
from data_utils import ( from lib.train.data_utils import (
TextAudioLoaderMultiNSFsid, TextAudioLoaderMultiNSFsid,
TextAudioLoader, TextAudioLoader,
TextAudioCollateMultiNSFsid, TextAudioCollateMultiNSFsid,
@ -45,9 +41,9 @@ else:
SynthesizerTrnMs768NSFsid_nono as RVC_Model_nof0, SynthesizerTrnMs768NSFsid_nono as RVC_Model_nof0,
MultiPeriodDiscriminatorV2 as MultiPeriodDiscriminator, MultiPeriodDiscriminatorV2 as MultiPeriodDiscriminator,
) )
from losses import generator_loss, discriminator_loss, feature_loss, kl_loss from lib.train.losses import generator_loss, discriminator_loss, feature_loss, kl_loss
from mel_processing import mel_spectrogram_torch, spec_to_mel_torch from lib.train.mel_processing import mel_spectrogram_torch, spec_to_mel_torch
from process_ckpt import savee from lib.train.process_ckpt import savee
global_step = 0 global_step = 0