From 27101d3d88176910d91e6582c64c945ce83938fd Mon Sep 17 00:00:00 2001
From: HydrogenSulfate <490868991@qq.com>
Date: Mon, 19 Sep 2022 14:37:15 +0800
Subject: [PATCH] fix bug for warmup_steps setting

---
 ppcls/optimizer/learning_rate.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/ppcls/optimizer/learning_rate.py b/ppcls/optimizer/learning_rate.py
index a2a850be..437ffe24 100644
--- a/ppcls/optimizer/learning_rate.py
+++ b/ppcls/optimizer/learning_rate.py
@@ -56,9 +56,8 @@ class LRBase(object):
         self.step_each_epoch = step_each_epoch
         self.learning_rate = learning_rate
         self.warmup_epoch = warmup_epoch
-        self.warmup_steps = round(
-            self.warmup_epoch *
-            self.step_each_epoch) if by_epoch else self.warmup_epoch
+        self.warmup_steps = self.warmup_epoch if by_epoch else round(
+            self.warmup_epoch * self.step_each_epoch)
         self.warmup_start_lr = warmup_start_lr
         self.last_epoch = last_epoch
         self.by_epoch = by_epoch
-- 
GitLab