v2

PlayVoice · Sep 20, 2023 · 23c96c3 · 23c96c3
1 parent e3d98ea
commit 23c96c3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/grad_extend/train.py b/grad_extend/train.py
@@ -45,6 +45,8 @@ def train(hps, chkpt_path=None):
         checkpoint = torch.load(hps.train.pretrain, map_location='cpu')
         load_model(model, checkpoint['model'])
         hps.train.learning_rate = 2e-5
+        # fine_tune
+        model.fine_tune()
     else:
         print_error(10 * '~' + "No Pretrain Model" + 10 * '~')
 
@@ -73,7 +75,7 @@ def train(hps, chkpt_path=None):
 
     print('Start training...')
     skip_diff_train = True
-    if initepoch > hps.train.fast_epochs:
+    if initepoch >= hps.train.fast_epochs:
         skip_diff_train = False
     for epoch in range(initepoch, hps.train.full_epochs + 1):