Upload train.py

train.py

@@ -9,7 +9,7 @@ sys.path.append(os.path.join(now_dir))
 
 import datetime
 
-from infer.lib.train import utils
+from lib.train import utils
 
 hps = utils.get_hparams()
 os.environ["CUDA_VISIBLE_DEVICES"] = hps.gpus.replace("-", ",")
@@ -45,8 +45,8 @@ from torch.nn.parallel import DistributedDataParallel as DDP
 from torch.utils.data import DataLoader
 from torch.utils.tensorboard import SummaryWriter
 
-from infer.lib import commons
-from infer.lib.data_utils import (
+from lib import commons
+from lib.data_utils import (
     DistributedBucketSampler,
     TextAudioCollate,
     TextAudioCollateMultiNSFsid,