入门客AI创业平台(我带你入门,你带我飞行)
博文笔记

Tensorflow1.5 可视化及tf.summary.merge_all()的坑

创建时间:2018-02-11 投稿人: 浏览次数:681

1.自动管理模式:

注意:自动管理模式下,导入已保存的模型继续训练时,会抛异常,该种场景下建议使用手动模式

summary_writer = tf.summary.FileWriter("E:/data/tensorflow-master/1.Cnn_Captcha/result/",  flush_secs=60)
summary_writer.add_graph(sess.graph)#添加graph图
tf.summary.scalar("loss", loss)
tf.summary.scalar("accuracy", accuracy)
sum_ops = tf.summary.merge_all()#自动管理
metall = sess.run(sum_ops, feed_dict={X: batch_x_test, Y: batch_y_test, keep_prob: 1.})
summary_writer.add_summary(metall, global_step=step)  # 写入文件

2.手动管理模式

summary_writer = tf.summary.FileWriter("E:/data/tensorflow-master/1.Cnn_Captcha/result/",  flush_secs=60)
summary_writer.add_graph(sess.graph)#添加graph图
loss_scalar = tf.summary.scalar("loss", loss)
accuracy_scalar = tf.summary.scalar("accuracy", accuracy)
loss_metall, accuracy_metall,  = sess.run([loss_scalar, accuracy_scalar], feed_dict={X: batch_x_test, Y: batch_y_test, keep_prob: 1.})
summary_writer.add_summary(loss_metall, global_step=step)  # 写入文件
summary_writer.add_summary(accuracy_metall, global_step=step)  # 写入文件

声明:该文观点仅代表作者本人,入门客AI创业平台信息发布平台仅提供信息存储空间服务,如有疑问请联系rumenke@qq.com。
  • 上一篇:没有了
  • 下一篇:没有了
未上传头像