6fe9fb323ba3292fb427173095f99669f832d836,examples/demo_graph.py,,,#,124

Before Change

        return x


dummy_input = Variable(torch.rand(13, 1, 28, 28))

model = Net1()
with SummaryWriter(comment="Net1") as w:
    w.add_graph(model, (dummy_input, ))

model = Net2()
with SummaryWriter(comment="Net2") as w:
    w.add_graph(model, (dummy_input, ))

dummy_input = torch.Tensor(1, 3, 224, 224)

with SummaryWriter(comment="alexnet") as w:
    model = torchvision.models.alexnet()

After Change

dummy_input = torch.rand(1, 3, 224, 224)

with SummaryWriter(comment="basicblock") as w:
    model = BasicBlock(3, 3)
    w.add_graph(model, (dummy_input, ), verbose=True)

Instances