We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 43028e1 commit a3ca96bCopy full SHA for a3ca96b
torchelie/lr_scheduler.py
@@ -31,7 +31,7 @@ def __init__(self,
31
if transition == 'log':
32
self.schedule = [(a, math.log(b), math.log(c))
33
for a, b, c in self.schedule]
34
- super().__init__(optimizer, last_epoch, verbose)
+ super().__init__(optimizer, last_epoch)
35
36
def step(self, *unused) -> None:
37
"""
@@ -113,7 +113,7 @@ def __init__(self,
113
verbose: bool = False):
114
self.total_iters = total_iters
115
self.warmup = warmup_ratio
116
117
118
119
@@ -248,7 +248,7 @@ def __init__(self,
248
self.n_iters_total = n_iters_total
249
self.tanh_lower_bound = tanh_lower_bound
250
self.tanh_upper_bound = tanh_upper_bound
251
252
253
def get_lr(self) -> List[float]:
254
t = self.last_epoch / self.n_iters_total
0 commit comments