擅长:python、mysql、java
<p>您可以创建一个从<a href="https://github.com/keras-team/keras/blob/9400be98783135a1d42dd238f4e6c3aa048eceea/keras/callbacks.py#L1061" rel="nofollow noreferrer">ReduceLROnPlateau</a>继承的自定义回调,如下所示:</p>
<pre><code>class CheckpointLR(ReduceLROnPlateau):
# override on_epoch_end()
def on_epoch_end(self, epoch, logs=None):
if not self.in_cooldown():
temp = self.model.get_weights()
self.model.set_weights(self.last_weights)
self.last_weights = temp
super().on_epoch_end(epoch, logs) # actually reduce LR
</code></pre>