Tensorboard

1、在训练模型中添加关于tensorboard的回调函数

tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir, histogram_freq=1)

2、定义存储目录

log_dir = os.path.join('logs', datetime.datetime.now().strftime("%Y%m%d-%H%M%S"))

3、在fit中添加

添加callbacks

model.fit(callbacks=[tensorboard_callback])

4、显示

4.1 在jupter lab中显示

%load_ext tensorboard
%tensorboard --logdir logs

4.2 在浏览器中显示

tensorboard --logdir logs