tensorflow.keras使用TensorBoard的方法

把原有的训练代码中的fit的那行代码改成下面的代码：

tbcb= TensorBoard(log_dir="./tb_model_save_dir", histogram_freq=1, write_grads=True)
model.fit(x_train, y_train, batch_size=batch_size, epochs=epochs, validation_split=0.1,callbacks=[tbcb])

训练完成之后。

在tb_model_save_dir文件夹的上一层文件夹中执行下面的命令：

tensorboard  --bind_all --logdir ./tb_model_save_dir

在浏览器中输入http://your_ip:6006/就可以看到自己的模型的信息。

使用TensorBoard可能会减慢训练的速度。