老师，能指导下这个写tensorboard吗

来源：6-20 TensorFlow-ssd模型训练之框架解读（2）

O_O_似水流年_O_O

2020-01-01

老师，对于ssd这个模型，我需要用tensorboard查看他的loss图，我不知道在哪里写怎么写，麻烦老师教一下

写回答

1回答

会写代码的好厨师

2020-01-31

已采纳

tensorboard的简单例子：

#这里使用tf.summary.* 在定义网络的时候添加相应的tensor，比如下面

tf.summary.scalar('accuracy',acc) #生成准确率标量图

##后面是在训练的时候添加的标准代码

merge_summary = tf.summary.merge_all()

train_writer = tf.summary.FileWriter(dir,sess.graph)

#每次训练在run的时候，执行merge_summary

for step in range(training_step): #训练循环

train_summary = sess.run(merge_summary,feed_dict = {...})

#调用sess.run运行图，生成一步的训练过程数据

train_writer.add_summary(train_summary, step)

Python3+TensorFlow打造人脸识别智能小程序

理论与实战项目双管齐下，让AI技术真正落地应用，适合毕设展示。

1086 学习 · 538 问题

相似问题

回答 2

回答 1

回答 2

回答 2

回答 1