Upload train.py

train.py

@@ -9,7 +9,7 @@ sys.path.append(os.path.join(now_dir))
 
 import datetime
 
-from lib.train import utils
+from lib import utils
 
 hps = utils.get_hparams()
 os.environ["CUDA_VISIBLE_DEVICES"] = hps.gpus.replace("-", ",")
@@ -67,14 +67,14 @@ else:
         MultiPeriodDiscriminatorV2 as MultiPeriodDiscriminator,
     )
 
-from infer.lib.train.losses import (
+from lib.losses import (
     discriminator_loss,
     feature_loss,
     generator_loss,
     kl_loss,
 )
-from infer.lib.train.mel_processing import mel_spectrogram_torch, spec_to_mel_torch
-from infer.lib.train.process_ckpt import savee
+from lib.mel_processing import mel_spectrogram_torch, spec_to_mel_torch
+from lib.process_ckpt import savee
 
 global_step = 0