diff --git a/train.py b/train.py index 0c63a514..24a2f7ab 100644 --- a/train.py +++ b/train.py @@ -140,7 +140,7 @@ def train(): # scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lf) # scheduler = lr_scheduler.MultiStepLR(optimizer, milestones=range(59, 70, 1), gamma=0.8) # gradual fall to 0.1*lr0 scheduler = lr_scheduler.MultiStepLR(optimizer, milestones=[round(epochs * x) for x in [0.8, 0.9]], gamma=0.1) - scheduler.last_epoch = start_epoch - 1 + scheduler.last_epoch = start_epoch # # Plot lr schedule # y = []