Skip to content

GitLab

Unverified Commit a2270633 authored by Yan haixu's avatar Yan haixu Committed by GitHub
Browse files

[misc] add last_epoch in CosineAnnealingWarmupLR (#4778)

parent b6cf0aca
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/nn/lr_scheduler/cosine.py
View file @ a2270633
......@@ -62,7 +62,7 @@ class CosineAnnealingWarmupLR(WarmupScheduler):
base_scheduler = _CosineAnnealingLR(
optimizer, total_steps - warmup_steps, eta_min=eta_min, last_epoch=last_epoch
)
super().__init__(optimizer, warmup_steps, base_scheduler)
super().__init__(optimizer, warmup_steps, base_scheduler, last_epoch=last_epoch)
class FlatAnnealingLR(DelayerScheduler):
......
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