Add Exponential LR scheduler check

2026-09-09 11:16:00 +08:00 · 2021-04-08 11:20:33 +02:00
parent 1535777f64
commit ef6ff4e95c
1 changed files with 2 additions and 0 deletions
@@ -506,6 +506,8 @@ def main(args):  # pylint: disable=redefined-outer-name
                print(" > Restoring Discriminator LR Scheduler...")
                scheduler_disc.load_state_dict(checkpoint['scheduler_disc'])
                scheduler_disc.optimizer = optimizer_disc
+                if c.lr_scheduler_disc == "ExponentialLR":
+                    scheduler_disc.last_epoch = checkpoint['epoch']
        except RuntimeError:
            # restore only matching layers.
            print(" > Partial model initialization...")