restore ckpt for disc too.

( Had it off for some testing.
codename0og · Dec 28, 2024 · 02e6ffe · 02e6ffe
1 parent 233b8c6
commit 02e6ffe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rvc/train/train.py b/rvc/train/train.py
@@ -394,7 +394,7 @@ def run(
         checkpointing=use_checkpointing
     ).to(device)
 
-    net_d = MultiPeriodDiscriminator(version, config.model.use_spectral_norm, False).to(device)
+    net_d = MultiPeriodDiscriminator(version, config.model.use_spectral_norm, use_checkpointing).to(device)
 
     optim_g = torch.optim.RAdam(
         net_g.parameters(),