I have made a UNet model for segmentation task. Which tool can i use for drawing it apart from graphviz. I tried it using nnsvg but i think it cannot make U-Net.
2)__ Netron can be used to produce interactive visualizations of neural network models. It is available as a Python package, a standalone application, and a browser application.