def update_learning_rate(self, lr):
if isinstance(self.optimizer, optimizers.Adam):
self.optimizer.alpha = lr
return
if isinstance(self.optimizer, Eve):
self.optimizer.alpha = lr
return
if isinstance(self.optimizer, optimizers.AdaDelta):
# AdaDelta has no learning rate
return
self.optimizer.lr = lr
评论列表
文章目录