From 85a467c3865d219760e2a429f0c32fae4a562fa3 Mon Sep 17 00:00:00 2001 From: OedoSoldier <31711261+OedoSoldier@users.noreply.github.com> Date: Mon, 2 Oct 2023 18:10:16 +0800 Subject: [PATCH] Update train_ms.py (#46) Fix error when resuming training --- train_ms.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/train_ms.py b/train_ms.py index 2dcef50..1f1708d 100644 --- a/train_ms.py +++ b/train_ms.py @@ -191,6 +191,8 @@ def run(): optim_g.param_groups[0]["initial_lr"] = g_resume_lr if not optim_d.param_groups[0].get("initial_lr"): optim_d.param_groups[0]["initial_lr"] = d_resume_lr + if not optim_dur_disc.param_groups[0].get("initial_lr"): + optim_dur_disc.param_groups[0]["initial_lr"] = dur_resume_lr epoch_str = max(epoch_str, 1) global_step = (epoch_str - 1) * len(train_loader)