Debug

2023-05-23 00:03:25 +08:00 · 2023-05-23 00:03:25 +08:00 · d7550743a3
parent dd8771c55e
commit d7550743a3
1 changed files with 2 additions and 3 deletions
--- a/train.py
+++ b/train.py
@ -137,9 +137,8 @@ def run(rank, n_gpus, hps):
            train_and_evaluate(rank, epoch, hps, [net_g, net_d], [optim_g, optim_d], [scheduler_g, scheduler_d], scaler,
                               [train_loader, None], None, None)
        # update learning rate
-        if epoch > 1:
-            scheduler_g.step()
-            scheduler_d.step()
+        scheduler_g.step()
+        scheduler_d.step()


 def train_and_evaluate(rank, epoch, hps, nets, optims, schedulers, scaler, loaders, logger, writers):