23.自定义训练中使用Tensorboard

由于我们自定义训练中没有model.fit(),所以按照之前的方法添加Tensorboard是做不到的

我们对 20.自定义训练(非编译训练) 的代码做修改

导入datetime库

定义current_time

定义train与test日志路径

然后创建train与test的写对象

之后修改我们的训练函数

训练之后打开tensorboard看一下,发现有我们保存的四个变量

  • acc

  • loss

  • val_acc

  • val_loss

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
TensorBoard 是一个由 TensorFlow 提供的可视化工具,可以帮助用户更好地理解和调试深度学习模型。使用 TensorBoard,您可以可视化训练过程的损失和指标、模型结构、梯度直方图、嵌入向量等等。 下面是 TensorBoard使用步骤: 1. 在训练脚本添加 TensorBoard 回调函数,例如: ``` from tensorflow.keras.callbacks import TensorBoard tensorboard_callback = TensorBoard(log_dir='./logs', histogram_freq=1) model.fit(x_train, y_train, epochs=10, callbacks=[tensorboard_callback]) ``` 其 `log_dir` 参数是指定 TensorBoard 日志文件的路径,`histogram_freq` 参数是指定记录梯度直方图的频率。 2. 在终端运行 TensorBoard 命令,例如: ``` tensorboard --logdir=./logs ``` 其 `--logdir` 参数是指定日志文件的路径。 3. 在浏览器打开 TensorBoard 界面,例如: ``` http://localhost:6006 ``` 其 `6006` 是默认的 TensorBoard 端口号,您也可以在运行 TensorBoard 命令时指定端口号。 在 TensorBoard 界面,您可以查看训练过程的损失和指标曲线、模型结构、梯度直方图、嵌入向量等等。如果您使用的是 TensorFlow 2.x 版本,还可以使用 `tf.summary` API 记录自定义的数据,例如: ``` with tf.summary.create_file_writer(log_dir).as_default(): tf.summary.scalar('loss', loss, step=step) tf.summary.scalar('accuracy', accuracy, step=step) ``` 使用 TensorBoard 可以更加直观地了解深度学习模型的训练过程,推荐大家在开发深度学习模型时使用

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Suyuoa

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值