Skip to content

P
PaddleClas

PaddlePaddle / PaddleClas
1 年多 前同步成功

通知 116
Star 4999
Fork 1114
P
PaddleClas
提交 57dcf7ad 编写于 作者: H HydrogenSulfate
浏览文件
操作

fix small bugs

上级 7e072087
隐藏空白更改
内联 并排
Showing with 11 addition and 11 deletion
ppcls/optimizer/learning_rate.py
浏览文件 @ 57dcf7ad
...@@ -116,9 +116,9 @@ class Constant(LRBase): ...@@ -116,9 +116,9 @@ class Constant(LRBase):
last_epoch=-1, last_epoch=-1,
by_epoch=False, by_epoch=False,
**kwargs): **kwargs):
super(Linear, self).__init__(epochs, step_each_epoch, learning_rate, super(Constant, self).__init__(epochs, step_each_epoch, learning_rate,
warmup_epoch, warmup_start_lr, last_epoch, warmup_epoch, warmup_start_lr,
by_epoch) last_epoch, by_epoch)
def __call__(self): def __call__(self):
learning_rate = lr.LRScheduler( learning_rate = lr.LRScheduler(
...@@ -220,7 +220,7 @@ class Cosine(LRBase): ...@@ -220,7 +220,7 @@ class Cosine(LRBase):
last_epoch=-1, last_epoch=-1,
by_epoch=False, by_epoch=False,
**kwargs): **kwargs):
super(Linear, self).__init__(epochs, step_each_epoch, learning_rate, super(Cosine, self).__init__(epochs, step_each_epoch, learning_rate,
warmup_epoch, warmup_start_lr, last_epoch, warmup_epoch, warmup_start_lr, last_epoch,
by_epoch) by_epoch)
self.T_max = (self.epochs - self.warmup_epoch) * self.step_each_epoch self.T_max = (self.epochs - self.warmup_epoch) * self.step_each_epoch
...@@ -269,9 +269,9 @@ class Step(LRBase): ...@@ -269,9 +269,9 @@ class Step(LRBase):
last_epoch=-1, last_epoch=-1,
by_epoch=False, by_epoch=False,
**kwargs): **kwargs):
super(Linear, self).__init__(epochs, step_each_epoch, learning_rate, super(Step, self).__init__(epochs, step_each_epoch, learning_rate,
warmup_epoch, warmup_start_lr, last_epoch, warmup_epoch, warmup_start_lr, last_epoch,
by_epoch) by_epoch)
self.step_size = step_size * step_each_epoch self.step_size = step_size * step_each_epoch
self.gamma = gamma self.gamma = gamma
if self.by_epoch: if self.by_epoch:
...@@ -315,7 +315,7 @@ class Piecewise(LRBase): ...@@ -315,7 +315,7 @@ class Piecewise(LRBase):
last_epoch=-1, last_epoch=-1,
by_epoch=False, by_epoch=False,
**kwargs): **kwargs):
super(Linear, super(Piecewise,
self).__init__(epochs, step_each_epoch, values[0], warmup_epoch, self).__init__(epochs, step_each_epoch, values[0], warmup_epoch,
warmup_start_lr, last_epoch, by_epoch) warmup_start_lr, last_epoch, by_epoch)
self.values = values self.values = values
...@@ -362,9 +362,9 @@ class MultiStepDecay(LRBase): ...@@ -362,9 +362,9 @@ class MultiStepDecay(LRBase):
last_epoch=-1, last_epoch=-1,
by_epoch=False, by_epoch=False,
**kwargs): **kwargs):
super(Linear, self).__init__(epochs, step_each_epoch, learning_rate, super(MultiStepDecay, self).__init__(
warmup_epoch, warmup_start_lr, last_epoch, epochs, step_each_epoch, learning_rate, warmup_epoch,
by_epoch) warmup_start_lr, last_epoch, by_epoch)
self.milestones = [x * step_each_epoch for x in milestones] self.milestones = [x * step_each_epoch for x in milestones]
self.gamma = gamma self.gamma = gamma
if self.by_epoch: if self.by_epoch:
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或