提交 db80bfb5 编写于 作者: H HydrogenSulfate 提交者: Tingquan Gao

change return type from float to Constant when self.learning_rate is an float number

上级 0160b8dd
...@@ -194,8 +194,8 @@ class Linear(LRBase): ...@@ -194,8 +194,8 @@ class Linear(LRBase):
end_lr=self.end_lr, end_lr=self.end_lr,
power=self.power, power=self.power,
cycle=self.cycle, cycle=self.cycle,
last_epoch=self. last_epoch=self.last_epoch) if self.decay_steps > 0 else Constant(
last_epoch) if self.decay_steps > 0 else self.learning_rate self.learning_rate)
if self.warmup_steps > 0: if self.warmup_steps > 0:
learning_rate = self.linear_warmup(learning_rate) learning_rate = self.linear_warmup(learning_rate)
...@@ -243,8 +243,8 @@ class Cosine(LRBase): ...@@ -243,8 +243,8 @@ class Cosine(LRBase):
learning_rate=self.learning_rate, learning_rate=self.learning_rate,
T_max=self.T_max, T_max=self.T_max,
eta_min=self.eta_min, eta_min=self.eta_min,
last_epoch=self. last_epoch=self.last_epoch) if self.T_max > 0 else Constant(
last_epoch) if self.T_max > 0 else self.learning_rate self.learning_rate)
if self.warmup_steps > 0: if self.warmup_steps > 0:
learning_rate = self.linear_warmup(learning_rate) learning_rate = self.linear_warmup(learning_rate)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册