Take a look at our example notebook, streamlit version runs here The main idea is for you do experiment and develop in your notebook, visually see all the pieces, and then convert the notebook to .py ...

If you installed GraphViz successfully, you can install the torchviz package. This package is not part of Anaconda Distribution Repository and is only available at PyPI , the Python Package Index, so ...