szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
MIT License
3.18k stars 279 forks source link

How to adjust the resolution of the saved dot image? #41

Open Jian-danai opened 4 years ago

Jian-danai commented 4 years ago

How to adjust the resolution of the saved dot image? My computational graph is too long...then the image seems not clear enough. image

ucalyptus commented 4 years ago

@Jian-danai very easy

def resize_graph(dot, size_per_element=0.15, min_size=12):
    """Resize the graph according to how much content it contains.
    Modify the graph in place.
    """
    # Get the approximate number of nodes and edges
    num_rows = len(dot.body)
    content_size = num_rows * size_per_element
    size = max(min_size, content_size)
    size_str = str(size) + "," + str(size)
    dot.graph_attr.update(size=size_str)

resize_graph(dot,size_per_element=1,min_size=20)
dot.format = 'png'
dot.render('hehe')
mkmohangb commented 2 years ago

Use svg format:

output  = model(inputs)
dot = make_dot(output).render("filename", format="svg")