Skip to content

Commit f409af4

Browse files
committed
add PolyScheduler
1 parent 74c93c8 commit f409af4

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

change_detection_pytorch/utils/lr_scheduler.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,10 +142,9 @@ def __init__(self,
142142
raise TypeError('{} is not an Optimizer'.format(
143143
type(optimizer).__name__))
144144
self.optimizer = optimizer
145+
self.power = power
145146
self.by_epoch = by_epoch
146-
self.epochs = epochs
147147
self.min_lr = min_lr
148-
self.power = power
149148

150149
# Validate total_steps
151150
if by_epoch:

0 commit comments

Comments
 (0)