From 0776f1c70030924f4a7973234d0dce7dc496d41c Mon Sep 17 00:00:00 2001
From: HydrogenSulfate <490868991@qq.com>
Date: Fri, 16 Sep 2022 14:40:51 +0800
Subject: [PATCH] fix small bugs

---
 ppcls/optimizer/learning_rate.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppcls/optimizer/learning_rate.py b/ppcls/optimizer/learning_rate.py
index c88c5ea76..a2a850be6 100644
--- a/ppcls/optimizer/learning_rate.py
+++ b/ppcls/optimizer/learning_rate.py
@@ -58,7 +58,7 @@ class LRBase(object):
         self.warmup_epoch = warmup_epoch
         self.warmup_steps = round(
             self.warmup_epoch *
-            self.step_each_epoch) if self.by_epoch else self.warmup_epoch
+            self.step_each_epoch) if by_epoch else self.warmup_epoch
         self.warmup_start_lr = warmup_start_lr
         self.last_epoch = last_epoch
         self.by_epoch = by_epoch