只需要在fit或fit_generator前加上这几句代码即可,log_filepath前要加r解析为路径,否则会报错
log_filepath = r'logs'
tb_cb = TensorBoard(log_dir=log_filepath, write_images=1, histogram_freq=1)
cbks = [tb_cb]
model.fit_generator(generate_one_epoch(train_X, train_y, batch_size=batch_size),
steps_per_epoch=len(train_X)//batch_size,
epochs=100, callbacks=cbks)
同时在当前目录下新建一个空的logs文件夹
网络开始训练后进入logs文件夹,如下图,在该页面开启终端窗口(shift+鼠标右键,“在此处打开PowerShell窗口”),输入命令 tensorboard --logdir=./,
出现下图,复制网址到浏览器打开即可