提交 0776f1c7 编写于 作者: H HydrogenSulfate

fix small bugs

上级 57dcf7ad
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册