Making Architecture Diagrams for neural networks

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.

01 January 1970 4,039 2 View