Merge pull request #9 from theojourdan/theojourdan-patch-1

Correction of load condition in training.py
galgreshler · Apr 2, 2024 · 18c5a64 · 18c5a64
2 parents 6abb81e + 9229c63
commit 18c5a64
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(