keras使用tensorboard
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# 导入库
from keras.callbacks import TensorBoard
# 创建tensorboard对象, 结果保存在logs目录下
tensorboard = TensorBoard(log_dir='logs/{}'.format(NAME),histogram_freq=1,write_grads=True)
# 在模型生成器函数作为回调参数

model.fit_generator(
generator=train_generator,
epochs=100,
steps_per_epoch=2200 // BATCH_SIZE,
validation_data=validation_generator,
validation_steps=200 // BATCH_SIZE,
callbacks=[tensorboard]
)

在浏览器中展示数据

1
tensorboard --logdir=logs
文章作者: Spaceack
文章链接: http://spaceack.com/2020/04/23/2020-04-23-keras%E4%BD%BF%E7%94%A8Tensorboard/
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 丸子家的小云吞
支付宝打赏
微信打赏