UML for deep learning architecture

username: Arthur_Zakirov
cooked: <p>Hello everyone,</p>
<p>can anybody share his experience with tools / methods for a clean documentation of Models?<br>
I’ve tried using class diagram for my NTM.<br>
<div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/6/5/6519c475e3cf834fb514631429322cab9c0f8de2.png" data-download-href="https://discuss.pytorch.org/uploads/default/6519c475e3cf834fb514631429322cab9c0f8de2" title="image"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/6/5/6519c475e3cf834fb514631429322cab9c0f8de2_2_690x214.png" alt="image" data-base62-sha1="eqnqIDlIkzhVrfICTUIWzvXRG9A" width="690" height="214" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/6/5/6519c475e3cf834fb514631429322cab9c0f8de2_2_690x214.png, https://discuss.pytorch.org/uploads/default/optimized/3X/6/5/6519c475e3cf834fb514631429322cab9c0f8de2_2_1035x321.png 1.5x, https://discuss.pytorch.org/uploads/default/optimized/3X/6/5/6519c475e3cf834fb514631429322cab9c0f8de2_2_1380x428.png 2x" data-dominant-color="F9F7F3"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">image</span><span class="informations">1448×451 24 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>
<p>Also I had the idea of drawing an activity diagram with a swimlane for each component to describe the flow of Tensors trough the model like this …<br>
<div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/e/d/edf99cf10ae38b15b8d206417f0acb4966390418.png" data-download-href="https://discuss.pytorch.org/uploads/default/edf99cf10ae38b15b8d206417f0acb4966390418" title="image"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/e/d/edf99cf10ae38b15b8d206417f0acb4966390418_2_615x499.png" alt="image" data-base62-sha1="xXdX99YwIFZ2nFGbY9zy33MxvCg" width="615" height="499" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/e/d/edf99cf10ae38b15b8d206417f0acb4966390418_2_615x499.png, https://discuss.pytorch.org/uploads/default/original/3X/e/d/edf99cf10ae38b15b8d206417f0acb4966390418.png 1.5x, https://discuss.pytorch.org/uploads/default/original/3X/e/d/edf99cf10ae38b15b8d206417f0acb4966390418.png 2x" data-dominant-color="F2F1F1"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">image</span><span class="informations">869×706 87.8 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>

username: marksaroufim
cooked: <p>There’s a bunch of open source tools <a href="https://github.com/szagoruyko/pytorchviz" class="inline-onebox" rel="noopener nofollow ugc">GitHub - szagoruyko/pytorchviz: A small package to create visualizations of PyTorch execution graphs</a></p>
<p>However if you have a better one in mind by all means create it, I think this is far from being a solved problem</p>