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

Before Change

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

with SummaryWriter(comment="vgg19") as w:
    model = torchvision.models.vgg19()
    w.add_graph(model, (dummy_input, ))

with SummaryWriter(comment="densenet121") as w:
    model = torchvision.models.densenet121()
    w.add_graph(model, (dummy_input, ))

with SummaryWriter(comment="resnet18") as w:
    model = torchvision.models.resnet18()
    w.add_graph(model, (dummy_input, ))

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