diff --git a/train_ms.py b/train_ms.py
index f76534d..c7547ae 100644
--- 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()