как использовать Tensorboard в Pytorch просто нарисуйте простую сеть - PullRequest
Новая
       15

как использовать Tensorboard в Pytorch просто нарисуйте простую сеть

1 голос
/ 01 февраля 2020

Я новый питчер. Я хочу нарисовать очень простой net только один вход с постоянным весом и смещением. Это net просто а = ш * х + б. Каждый вход скалярный. Я хочу использовать pytorch и тензорную доску, чтобы показать это net.

Я пишу код ниже. 

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()

enter image description here enter image description here

Я хочу, чтобы net не был таким, в чем проблема

...