Skip to content

Commit a3ca96b

Browse files
authored
Update lr_scheduler.py to latest pytorch
1 parent 43028e1 commit a3ca96b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

torchelie/lr_scheduler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def __init__(self,
3131
if transition == 'log':
3232
self.schedule = [(a, math.log(b), math.log(c))
3333
for a, b, c in self.schedule]
34-
super().__init__(optimizer, last_epoch, verbose)
34+
super().__init__(optimizer, last_epoch)
3535

3636
def step(self, *unused) -> None:
3737
"""
@@ -113,7 +113,7 @@ def __init__(self,
113113
verbose: bool = False):
114114
self.total_iters = total_iters
115115
self.warmup = warmup_ratio
116-
super().__init__(optimizer, last_epoch, verbose)
116+
super().__init__(optimizer, last_epoch)
117117

118118
def step(self, *unused) -> None:
119119
"""
@@ -248,7 +248,7 @@ def __init__(self,
248248
self.n_iters_total = n_iters_total
249249
self.tanh_lower_bound = tanh_lower_bound
250250
self.tanh_upper_bound = tanh_upper_bound
251-
super().__init__(optimizer, last_epoch, verbose)
251+
super().__init__(optimizer, last_epoch)
252252

253253
def get_lr(self) -> List[float]:
254254
t = self.last_epoch / self.n_iters_total

0 commit comments

Comments
 (0)