微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

在TensorFlow Keras API中,如何向CSVLogger回调创建的csv文件中添加参数?

如何解决在TensorFlow Keras API中,如何向CSVLogger回调创建的csv文件中添加参数?

我如何创建一个TensorFlow Keras API回调,以便在每个时期将学习率值添加到tf.keras.callbacks.CSVLogger创建的csv文件中?

通过下面的回调,我可以在每个时期之后打印出我的学习率,并将其添加到历史记录中。但是我不知道如何将其添加到CSVLogger csv文件中。似乎有一个存储在CSV记录器中的值存储在地方的日志字典,但我不知道它在哪里或如何添加

class Print_lr(tf.keras.callbacks.Callback):
    def on_epoch_begin(self,epoch,logs=None):
        print('lr = %f' % self.model.optimizer.lr)
        if 'lr' not in self.model.history.history.keys():
            self.model.history.history['lr'] = []
        self.model.history.history['lr'].append(self.model.optimizer.lr.numpy())

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。