mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

fix small bugs

Browse Source
This commit is contained in:
HydrogenSulfate 2022-09-16 14:40:51 +08:00
parent 57dcf7adb5
commit 0776f1c700
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/optimizer/learning_rate.py
View File

@ -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