From 143087b8accd80934e1a3cab1003cd5f7d619499 Mon Sep 17 00:00:00 2001
From: yukavio <67678385+yukavio@users.noreply.github.com>
Date: Thu, 7 Jan 2021 22:10:00 +0800
Subject: [PATCH] fix lr (#593)

---
 demo/prune/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/demo/prune/train.py b/demo/prune/train.py
index 44082c95..b85b7c7e 100644
--- a/demo/prune/train.py
+++ b/demo/prune/train.py
@@ -81,7 +81,7 @@ def piecewise_decay(args):
 def cosine_decay(args):
     step = int(math.ceil(float(args.total_images) / args.batch_size))
     learning_rate = paddle.optimizer.lr.CosineAnnealingDecay(
-        learning_rate=args.lr, T_max=args.num_epochs)
+        learning_rate=args.lr, T_max=args.num_epochs * step)
     optimizer = paddle.optimizer.Momentum(
         learning_rate=learning_rate,
         momentum=args.momentum_rate,
-- 
GitLab