保存日志问题
tensorflow吧
全部回复
仅看楼主
level 1
TTXYLX
楼主
summary_writer = tf.summary.FileWriter(log_save_path, sess.graph)
利用 gpu 在训练时,会生成一个空文件,0 kb,
模型可以保存,
利用 cpu 训练时,可以正常保存,
这是为什么
2019年04月08日 05点04分
1
level 1
TTXYLX
楼主
summary_writer = tf.summary.FileWriter(log_save_path, sess.graph)
summary_writer.add_summary(train_summary, ep)
summary_writer.add_summary(test_summary, ep)
summary_writer.flush()
2019年04月08日 05点04分
2
1