Graph View

Graph View module for pytorch model visualization.

visualtorch.graph.graph_view(model, input_shape, input_dtype=None, to_file=None, color_map=None, palette='okabe_ito', node_size=50, background_fill='white', padding=10, layer_spacing=250, node_spacing=10, type_ignore=None, outline_width=1, connector_fill='gray', connector_width=1, ellipsize_after=10, show_neurons=True, opacity=255, show_dimension=False, font=None, font_color='black', level_gap=None, show_input=True, show_arrows=False)

Generates an architecture visualization for a given linear PyTorch model in a graph style.

Return type:

Image

Deprecated:

Since version 1.4.0. Use visualtorch.render(model, input_shape, style=”graph”, …) instead - every parameter here is forwarded unchanged.