Update train.py

ziyaad30 · Oct 3, 2023 · 89cfcd9 · 89cfcd9
1 parent 560c6d6
commit 89cfcd9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train.py b/train.py
@@ -247,6 +247,7 @@ def run(rank, n_gpus, hps):
                     net_dur_disc,
                     optim_dur_disc,
                 )
+            epoch_str = 1
             global_step = step
 
         except: