panbaoran913 2022-07-23 22:28 采纳率: 71.4%
浏览 78
已结题

tensorboard,summarywriter

tensorboard不能正常创建文件的内容,打不开tensorboardX

当下正在学习torch下的SummaryWriter,记录loss损失。结果不能log下不能创建v2文件。创建的是0文件且有内容。

img

整体的代码过程
from tqdm import trange
import numpy as np
from torch.utils.tensorboard import SummaryWriter
import torch
import torch.nn as nn
import torch.optim as optim
device = torch.device('cpu')
net = nn.Linear(1,1).to(device) # 最简单全链接模型
loss_fn =nn.MSELoss().to(device)# 损失函数
optimizer = optim.SGD(net.parameters(),lr=0.001)
# 1. 输入数据
x_train = np.linspace(-1, 1, 100).reshape(100, 1)
y_train = 3 * np.power(x_train, 2) + 2 + 0.2 * np.random.randn(100, 1)
# 2. 实例化sw
sw1 = SummaryWriter(log_dir='log1')
# 3. 模型训练
for epoch in trange(40):
    inputs = torch.from_numpy(x_train).type(torch.float32).to(device)  # 所有输入
    targets = torch.from_numpy(y_train).type(torch.float32).to(device)  # 标签

    output = net(inputs)  # 模型的输入输出
    loss = loss_fn(output, targets)  # 预测则和真实值的loss

    loss.backward()  # 模型向后传递
    optimizer.step()  # 优化器优化

    sw1.add_scalar("train_loss", loss.item(), epoch)
sw1.close()
运行结果及报错内容

img


提示内容为

W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found
I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
打开tensorBordx的过程

img


复制: http://localhost:6006/到谷歌浏览器后,打开界面为

img

我想要正确记录loss值,并可以用网页打开
  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 7月31日
    • 创建了问题 7月23日

    悬赏问题

    • ¥15 is not in the mmseg::model registry。报错,模型注册表找不到自定义模块。
    • ¥15 安装quartus II18.1时弹出此error,怎么解决?
    • ¥15 keil官网下载psn序列号在哪
    • ¥15 想用adb命令做一个通话软件,播放录音
    • ¥30 Pytorch深度学习服务器跑不通问题解决?
    • ¥15 部分客户订单定位有误的问题
    • ¥15 如何在maya程序中利用python编写领子和褶裥的模型的方法
    • ¥15 Bug traq 数据包 大概什么价
    • ¥15 在anaconda上pytorch和paddle paddle下载报错
    • ¥25 自动填写QQ腾讯文档收集表