TensorBoard使用

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
在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章