X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=main.py;h=18c0730d7608866041a35874da1c1dc0b641133f;hb=a3abd0f58cfb2f2448c82db836093d20dc2954f2;hp=fabebddcffc31b5c383f2cacde6b8016e03ea1c5;hpb=664758db86b059b68cd11e889a20cc9681e4324a;p=mygptrnn.git

diff --git a/main.py b/main.py
index fabebdd..18c0730 100755
--- a/main.py
+++ b/main.py
@@ -478,7 +478,7 @@ def get_lr(n_epoch, it):
 
         if it < args.nb_warmup_iter:
             return args.legacy_large_lr * it / args.nb_warmup_iter
-        elif it < args.legacy_nb_epoch_large_lr:
+        elif n_epoch < args.legacy_nb_epoch_large_lr:
             return args.legacy_large_lr
         else:
             return args.legacy_small_lr