如何在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)

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

  • 6
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 22
    评论
评论 22
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值