TensorBoard使用

最新推荐文章于 2025-05-04 08:58:04 发布

enjoy_learn

最新推荐文章于 2025-05-04 08:58:04 发布

阅读量377

点赞数 2

CC 4.0 BY-SA版权

分类专栏： machine learning

本文链接：https://blog.youkuaiyun.com/enjoy_learn/article/details/98992678

machine learning 专栏收录该内容

4 篇文章

订阅专栏

from keras.callbacks import TensorBoard, ModelCheckpoint
from datetime import datetime
TIMESTAMP = "{0:%Y-%m-%dT%H-%M-%S/}".format(datetime.now())
train_log_dir = '/home/Predict/logs/train/' + TIMESTAMP
#print(train_log_dir)
tensorboard_callback = TensorBoard(log_dir=train_log_dir, histogram_freq=0, write_grads=False, batch_size=128,write_graph=True, write_images=False, embeddings_freq=0, embeddings_layer_names=None, embeddings_metadata=None)
callbacks_list = [tensorboard_callback]

在代码对应位置建立logs/train文件夹，然后右键启动终端输入：
tensorboard --logdir=logs
在这里插入图片描述