Update train_ms.py

2023-12-20 02:56:32 +08:00
parent 2e7c173de2
commit bd4591aa2c
1 changed files with 1 additions and 0 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -387,6 +387,7 @@ def run():
            )
        scheduler_g.step()
        scheduler_d.step()
+        scheduler_wd.step()
        if net_dur_disc is not None:
            scheduler_dur_disc.step()