Skip to content

GitLab

Commit 2b49ca80 authored by yuxuan-lou's avatar yuxuan-lou Committed by Frank Lee
Browse files

[NFC] polish colossalai/nn/lr_scheduler/linear.py code style (#1716)

parent f6389d08
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
colossalai/nn/lr_scheduler/linear.py
View file @ 2b49ca80
...@@ -24,5 +24,5 @@ class LinearWarmupLR(_LRScheduler): ...@@ -24,5 +24,5 @@ class LinearWarmupLR(_LRScheduler):
if self.last_epoch < self.warmup_steps: if self.last_epoch < self.warmup_steps:
return [(self.last_epoch + 1) / (self.warmup_steps + 1) * lr for lr in self.base_lrs] return [(self.last_epoch + 1) / (self.warmup_steps + 1) * lr for lr in self.base_lrs]
else: else:
return [(self.total_steps - self.last_epoch) / (self.total_steps - self.warmup_steps) * lr for lr in return [(self.total_steps - self.last_epoch) / (self.total_steps - self.warmup_steps) * lr
self.base_lrs] for lr in self.base_lrs]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment