Skip to content

Commit

Permalink
fix: little README.md clarification
Browse files Browse the repository at this point in the history
  • Loading branch information
MLRichter committed Jan 12, 2022
1 parent 936dcac commit bb68404
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Expand Up @@ -72,7 +72,7 @@ model = torchvision.models.alexnet()
graph = create_graph_from_pytorch_model(model)
visualize_architecture(
graph, f"alexnet_32_pixel", input_res=32
).render(f"alexnet_32_pixel")
).view()
```

This will create a graph of your model and visualize it using GraphViz and color all layers that are predicted to be
Expand Down

0 comments on commit bb68404

Please sign in to comment.