[QST] Visualize the NN architecture

[vivpra89]

❓ Questions & Help

Details

Is there an example on how to visualize the NN architecture after we get "model = tr.Model(head)". Also is it possible to extract it as a pytorch or TF model to use utils to plot.

[vivpra89]

from torchview import draw_graph

model_graph = draw_graph(model_si,input_data=batch) model_graph.resize_graph(scale=50)

displayHTML(model_graph.visual_graph.pipe(format="svg").decode("utf-8")).to_html()

[vivpra89]