Skip to content

Commit d43ce29

Browse files
Removed erroneous scale_fn from CLR subclasses (#1185)
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent 2ce92ab commit d43ce29

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

tensorflow_addons/optimizers/cyclical_learning_rate.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,6 @@ def __init__(
117117
initial_learning_rate: Union[FloatTensorLike, Callable],
118118
maximal_learning_rate: Union[FloatTensorLike, Callable],
119119
step_size: FloatTensorLike,
120-
scale_fn: Callable,
121120
scale_mode: str = "cycle",
122121
name: str = "TriangularCyclicalLearningRate",
123122
):
@@ -154,7 +153,6 @@ def __init__(
154153
a Python number. The maximum learning rate.
155154
step_size: A scalar `float32` or `float64` `Tensor` or a
156155
Python number. Step size.
157-
scale_fn: A function. Scheduling function applied in cycle
158156
scale_mode: ['cycle', 'iterations']. Mode to apply during cyclic
159157
schedule
160158
name: (Optional) Name for the operation.
@@ -180,7 +178,6 @@ def __init__(
180178
initial_learning_rate: Union[FloatTensorLike, Callable],
181179
maximal_learning_rate: Union[FloatTensorLike, Callable],
182180
step_size: FloatTensorLike,
183-
scale_fn: Callable,
184181
scale_mode: str = "cycle",
185182
name: str = "Triangular2CyclicalLearningRate",
186183
):
@@ -217,7 +214,6 @@ def __init__(
217214
a Python number. The maximum learning rate.
218215
step_size: A scalar `float32` or `float64` `Tensor` or a
219216
Python number. Step size.
220-
scale_fn: A function. Scheduling function applied in cycle
221217
scale_mode: ['cycle', 'iterations']. Mode to apply during cyclic
222218
schedule
223219
name: (Optional) Name for the operation.
@@ -243,7 +239,6 @@ def __init__(
243239
initial_learning_rate: Union[FloatTensorLike, Callable],
244240
maximal_learning_rate: Union[FloatTensorLike, Callable],
245241
step_size: FloatTensorLike,
246-
scale_fn: Callable,
247242
scale_mode: str = "iterations",
248243
gamma: FloatTensorLike = 1.0,
249244
name: str = "ExponentialCyclicalLearningRate",
@@ -282,7 +277,6 @@ def __init__(
282277
a Python number. The maximum learning rate.
283278
step_size: A scalar `float32` or `float64` `Tensor` or a
284279
Python number. Step size.
285-
scale_fn: A function. Scheduling function applied in cycle
286280
scale_mode: ['cycle', 'iterations']. Mode to apply during cyclic
287281
schedule
288282
gamma: A scalar `float32` or `float64` `Tensor` or a

0 commit comments

Comments
 (0)