fix: pre-trained model

infer/modules/train/train.py

@@ -95,17 +95,17 @@ def train(exp_dir: str):
 
     hps = utils.get_hparams_from_dir(exp_dir)
     hps.experiment_dir = exp_dir
-    hps.save_every_epoch = False
+    hps.save_every_epoch = 5
     hps.name = os.path.basename(exp_dir)
     hps.total_epoch = 100
-    hps.pretrainG = ""
-    hps.pretrainD = ""
+    hps.pretrainG = "assets/pretrained_v2/f0G40k.pth"
+    hps.pretrainD = "assets/pretrained_v2/f0D40k.pth"
     hps.version = "v2"
     hps.gpus = "0"
     hps.train.batch_size = 8
     hps.sample_rate = "40k"
     hps.if_f0 = 1
-    hps.if_latest = 1
+    hps.if_latest = 0
     hps.save_every_weights = "0"
     hps.if_cache_data_in_gpu = True
     hps.data.training_files = "%s/filelist.txt" % exp_dir
@@ -629,5 +629,3 @@ def train_and_evaluate(
                 )
             )
         )
-        sleep(1)
-        os._exit(2333333)