how can I show graph if don't use jupyter notebook

waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.

MIT License

1.79k stars 266 forks source link

how can I show graph if don't use jupyter notebook #13

Closed wangce888 closed 5 years ago

geekac commented 5 years ago

model = torchvision.models.vgg16() hl_graph = hl.build_graph(model, torch.zeros([1, 3, 224, 224])) hl_graph.save("xxx", format="png")

philferriere commented 5 years ago

Hi @aikudexiaohai,

If the information provided above doesn't answer your question or you're still facing issues, feel free to add more details to this thread and re-open this issue.

Thanks!