Pass figure size as argument for displaying huge graphs

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs

MIT License

3.24k stars 279 forks source link

Open kiranvaidhya opened 6 years ago

kiranvaidhya commented 6 years ago

Hi,

Can the output figure size be set as an argument instead of hard-coding it to (12, 12) here?

It helps when printing huge graphs.

szagoruyko commented 6 years ago

@kvrd18 sure, it's a good idea, can you send a PR?