@@ -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