From 9229c63a2018c048cdf571e41a2588729a7dd5fe Mon Sep 17 00:00:00 2001
From: Theo Jourdan <112558381+theojourdan@users.noreply.github.com>
Date: Tue, 2 Apr 2024 15:45:25 +0200
Subject: [PATCH] Update training.py

Wrong condition for loading previous model
---
 training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/training.py b/training.py
index a1862c0..78e67e9 100644
--- 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(