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