如何在kaggle上使用tensorboard

tensorflow官网上的教程
youtube上的教程

本教程结合以上两个教程得以实现在kaggle上使用tensorboard

以下代码都是在train_step的上面写的

log_dir="runs/"

summary_writer = tf.summary.create_file_writer(
  log_dir + "fit/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S"))

下载一个包

恩因为这里要下包,所以你的设置的时候要把联网打开,就是下面那个Internet
在这里插入图片描述

!wget https://bin.equinox.io/c/4VmDzA7iaHb/ngrok-stable-linux-amd64.zip
!unzip ngrok-stable-linux-amd64.zip

如果不行可以试下这个:
“现在ngrok要token了,在!unpzip ngrok后一行加入! “
./ngrok authtoken [你的token]
(评论区补充)

import os
import multiprocessing

pool = multiprocessing.Pool(processes = 10)
results_of_processes = [pool.apply_async(os.system, args=(cmd, ), callback = None )
                        for cmd in [
                        f"tensorboard --logdir ./runs/ --host 0.0.0.0 --port 6006 &",
                        "./ngrok http 6006 &"
                        ]]
! curl -s http://localhost:4040/api/tunnels | python3 -c \
    "import sys, json; print(json.load(sys.stdin)['tunnels'][0]['public_url'])"

在上面这段代码的下面会出现一个链接,你点进去就会到tensorboard的界面了。

接下来这段代码是在train_step里面写的

    with summary_writer.as_default():
        tf.summary.scalar('gen_loss', gen_total_loss, step=EPOCHS)
        tf.summary.scalar('disc_loss', disc_total_loss, step=EPOCHS)

就是添加你需要看的东西的曲线的信息

  • 5
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 22
    评论
Kaggle 平台上使用 TensorBoard 可以帮助您可视化和分析训练过程中的指标、损失函数值以及模型的结构等。Kaggle 提供了一种简单的方法来在 Kaggle Notebooks 中使用 TensorBoard。 下面是一个在 Kaggle Notebooks 中使用 TensorBoard 的示例代码: ```python from torch.utils.tensorboard import SummaryWriter # 创建 SummaryWriter 对象,将 TensorBoard 日志文件保存在默认路径 writer = SummaryWriter() # 记录训练过程中的指标 for epoch in range(num_epochs): # 计算指标 train_loss = ... train_accuracy = ... # 将指标写入 SummaryWriter writer.add_scalar('Train/Loss', train_loss, epoch) writer.add_scalar('Train/Accuracy', train_accuracy, epoch) # 关闭 SummaryWriter writer.close() ``` 运行以上代码后,会在当前目录下生成一个名为 `runs` 的文件夹,其中包含了 TensorBoard 日志文件。您可以在 Kaggle Notebooks 中使用以下代码来启动 TensorBoard: ```python %load_ext tensorboard %tensorboard --logdir runs ``` 然后,您可以在 Kaggle Notebooks 的输出窗口中看到一个链接,点击该链接即可打开 TensorBoard 界面。在 TensorBoard 中,您可以查看训练曲线、损失函数值、模型图表等,并进行比较和分析。 请注意,在 Kaggle 平台上使用 TensorBoard 需要确保您的环境支持 TensorBoard,并且已经安装了相关的库。您可以在 Kaggle Notebooks 中使用 `!pip install` 命令来安装所需的库。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值