From 13d63469b0a84ace0dc8848df47dc20538b98770 Mon Sep 17 00:00:00 2001
From: 34j <55338215+34j@users.noreply.github.com>
Date: Sun, 9 Apr 2023 19:42:18 +0900
Subject: [PATCH] fix(train): fix optim_d functions called in wrong order
 (#269)

---
 src/so_vits_svc_fork/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/so_vits_svc_fork/train.py b/src/so_vits_svc_fork/train.py
index 90c6a8d8..67800789 100644
--- a/src/so_vits_svc_fork/train.py
+++ b/src/so_vits_svc_fork/train.py
@@ -343,8 +343,8 @@ def training_step(self, batch: dict[str, torch.Tensor], batch_idx: int) -> None:
 
         # optimizer
         self.manual_backward(loss_disc_all)
-        optim_d.zero_grad()
         optim_d.step()
+        optim_d.zero_grad()
         self.untoggle_optimizer(optim_d)
 
     def validation_step(self, batch, batch_idx):