/Python/Learningrate scheduler tensorflow

Learningrate scheduler tensorflow

tf.keras.callbacks.LearningRateScheduler(
    schedule, verbose=0)

# This function keeps the initial learning rate for the first ten epochs
# and decreases it exponentially after that.

def scheduler(epoch, lr):
  if epoch < 10:
    return lr
  else:
    return lr * tf.math.exp(-0.1)

model = tf.keras.models.Sequential([tf.keras.layers.Dense(10)])
model.compile(tf.keras.optimizers.SGD(), loss='mse')

callback = tf.keras.callbacks.LearningRateScheduler(scheduler)
history = model.fit(np.arange(100).reshape(5, 20), np.zeros(5),
                    epochs=15, callbacks=[callback], verbose=0)

Similar codes

2 votes142 views2 answers
Count consecutive values in python
Python
0 vote127 views1 answer
How to convert csv columns to text python
Python
30 votes150 views6 answers
Get python version in code
Python
3 votes210 views4 answers
Seaborn correlation
Python
0 vote141 views1 answer
List Comprehension iteration
Python
0 vote135 views2 answers
Tkinter change ttk button color
Python
-11 votes172 views9 answers
çeviri
Python
0 vote137 views1 answer
Dataclass in python
Python
0 vote134 views1 answer
Python pickle 'ascii' codec can't decode byte 0x80 in position 0
Python
0 vote145 views2 answers
Compiling python code
Python