pytorch - 如何在 Pytorch 中使用 Tensorboard 只需画一个简单的网络

Question

我是一个新的pytorcher。我想画一个非常简单的网络，只有一个具有恒定权重和偏差的输入。这个网只有 a = w * x + b。每个输入都是标量。我想使用 pytorch 和 tensorboard 来展示这个网络。

我在下面写代码。

class Net(nn.Module):
    def __init__(self, w, b):
        super(Net, self).__init__()
        self.w = w
        self.b = b

    def forward(self, x):
        x = self.w * x + self.b
        return x


net = Net(1.3, 3.0)

writer = SummaryWriter('runs/CH1_2_1')
input = torch.tensor([1.0], dtype=torch.float)
input.requires_grad_(True)
writer.add_graph(net, input)
writer.close()

我要的网不应该是这样有什么问题

标签： pytorchtensorboard