只需要在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=./, 出现下图,复制网址到浏览器打开即可