tensorboard
使用:
(1) 程序中加入代码:
# 地址为自定义日志保存地址,空文件夹即可
# 后面的是模型名称
callbacks = [keras.callbacks.TensorBoard(log_dir='E:/lock/dir/'.format(model))]
callbacks = callbacks
如:
callbacks = [keras.callbacks.TensorBoard(log_dir='E:/lock/dir/'.format(model))]
# 开始训练
history = model.fit_generator(generator(path1, path2,BATCH_SIZE)
,steps_per_epoch=600, nb_epoch=EPOCH,callbacks = callbacks)
然后开始训练模型,接着日志目录下就会出现一个日志文件:
(2)cmd
执行:
tensorboard --logdir=E:/lock/dir/ --host=127.0.0.1 --port=8000
(注:几个都不要加单/双引号)
如果报错:not a tbloader or tbplugin subclass
解决:
执行pip list
执行pip uninstall tensorboard-plugin-wit
解决方法参考自:https://www.cnblogs.com/Dai-py/p/13271558.html
继续上面的步骤,显示类似下方代码就可以了:
接着浏览器打开这个网址:
但是数据没有正常显示,什么也不显示,想找原因也不知道怎么办,可能是谷歌浏览器的问题,换360浏览器
试试:
果然是浏览器问题。。。
如果显示如下信息,可能是日志还没有加载出来,一般要有个1-2
个epoch
以后才会显示。如果等待一段时间还是不行,首先检查日志文件的路径,其次如果是新版的keras
或tensorflow
,把命令中的等号去掉改为空格,并把几个参数用引号括起,参考:https://www.cnblogs.com/ipersevere/p/10843115.html