Update training.py

Wrong condition for loading previous model
galgreshler · Apr 2, 2024 · 9229c63 · 9229c63
1 parent 6abb81e
commit 9229c63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/training.py b/training.py
@@ -109,7 +109,7 @@ def train_single_scale(params, signals_list, fs_list, generators_list, noise_amp
 
     reconstruction_noise = signal_padder(reconstruction_noise)
 
-    if scale_idx > 1:
+    if scale_idx >= 1:
         netG.load_state_dict(
             torch.load('%s/netGScale%d.pth' % (params.output_folder, scale_idx - 1), map_location=params.device))
         netD.load_state_dict(