From 2b49ca80a30aad4decc3180c92cfbab118fd4a6e Mon Sep 17 00:00:00 2001 From: yuxuan-lou <83441848+yuxuan-lou@users.noreply.github.com> Date: Mon, 17 Oct 2022 17:51:40 +0800 Subject: [PATCH] [NFC] polish colossalai/nn/lr_scheduler/linear.py code style (#1716) --- colossalai/nn/lr_scheduler/linear.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/colossalai/nn/lr_scheduler/linear.py b/colossalai/nn/lr_scheduler/linear.py index 38992edf1..556938b8a 100644 --- a/colossalai/nn/lr_scheduler/linear.py +++ b/colossalai/nn/lr_scheduler/linear.py @@ -24,5 +24,5 @@ class LinearWarmupLR(_LRScheduler): if self.last_epoch < self.warmup_steps: return [(self.last_epoch + 1) / (self.warmup_steps + 1) * lr for lr in self.base_lrs] else: - return [(self.total_steps - self.last_epoch) / (self.total_steps - self.warmup_steps) * lr for lr in - self.base_lrs] + return [(self.total_steps - self.last_epoch) / (self.total_steps - self.warmup_steps) * lr + for lr in self.base_lrs]