码迷,mamicode.com
首页 > 其他好文 > 详细

keras 在train_on_batch中启用tensorboard

时间:2019-06-04 19:21:05      阅读:206      评论:0      收藏:0      [点我收藏+]

标签:pre   llb   value   for   zip   class   write   rbo   color   

def write_log(callback, names, logs, batch_no):
    for name, value in zip(names, logs):
        summary = tf.Summary()
        summary_value = summary.value.add()
        summary_value.simple_value = value
        summary_value.tag = name
        callback.writer.add_summary(summary, batch_no)
        callback.writer.flush()

在训练的时候,调用上述函数,写入tensorboard

  loss = model.train_on_batch([x1,x2],y)
  write_log(tensorboard_cb,["trainloss","me"],loss,bathNo)

 

keras 在train_on_batch中启用tensorboard

标签:pre   llb   value   for   zip   class   write   rbo   color   

原文地址:https://www.cnblogs.com/superxuezhazha/p/10975457.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!