tensorboardx可视画输出loss曲线用来分析过拟合,代码

安装tensorboardx
安装步骤:pip install tensorboardX

loss输出代码:

下面的参考:连接

#step1:倒入SummaryWriter
from tensorboardX import SummaryWriter
import numpy as np
#step:2初始化一个实例writer
writer = SummaryWriter('logs')#参数为指定存储路径

for i in range(100):
    #step:3记录loss
    #当数据只需要存储一个时用add_scalar()
    writer.add_scalar("test/sin",np.sin(i),i)#将我们所需要的数据保存在文件里面供可视化使用
    #当数据不止一个时使用add_scalars()
    writer.add_scalars("test1",{"sin":np.sin(i),"cos":np.cos(i)},i)
#step4:close
writer.close()

以上代码中,输出loss时,for循环中的i 就相当于那个epoch,
最关键的是writer.add_scalars(“test1”,{“sin”:np.sin(i),“cos”:np.cos(i)},i)
对应输出loss的代码是writer.add_scalars(args.result_path + ‘Train_val_loss’, {args.result_path + ‘train_loss’: total_loss.data.item()},epoch),
其中args.result_path + 'Train_val_loss’是输出文件的保存地址,args.result_path + 'train_loss’是线的名称,total_loss就是损失值loss了,也就是说{}里面的就是y值,epoch是横轴

将结果呈现出来:
终端输入tensorboard --logdir= “logs” # 绝对路径比较保险可能
其中logs是tensorboard输出文件的地址,不加“ ”也可以
如果没有呈现出来。就将命令行中的网址http://localhost:6006/复制到浏览器中,即可可视化
在这里插入图片描述
在这里插入图片描述

需要注意的是,像下面这样就可能会运行错误的,
writer.add_scalars(args.result_path + ‘loss’, {“train_loss”: total_loss.data.item(),“eval_loss”: eval_loss.data.item()},epoch)
原因是数据类型错误


网络断开,让远程服务器程序继续执行的命令
nohup 你的命令 > myout.file 2>&1 &
注意2不要省略,还错就是你的命令错误了

  • 0
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值