我们想要追踪模型的训练过程,tensorboard 是个不错的可视化工具。
安装tensorboard
还是使用conda虚拟环境,没有的话先安装conda,参考这里
conda install tensorboard
训练模型时调用callback API
训练信息会保存在 os.path.join(modelfile,‘logs’+model.name) 文件夹里
tensorboard_callback = keras.callbacks.TensorBoard(log_dir=os.path.join(modelfile,'logs'+model.name))
history = model.fit(x_tra, y_tra