{"id":15065069,"url":"https://github.com/mert-kurttutan/torchview","last_synced_at":"2025-10-23T18:37:37.618Z","repository":{"id":63069861,"uuid":"563575902","full_name":"mert-kurttutan/torchview","owner":"mert-kurttutan","description":"torchview: visualize pytorch models","archived":false,"fork":false,"pushed_at":"2025-04-22T09:39:06.000Z","size":42517,"stargazers_count":908,"open_issues_count":27,"forks_count":43,"subscribers_count":9,"default_branch":"main","last_synced_at":"2025-04-22T10:59:04.294Z","etag":null,"topics":["deep-learning","keras","machine-learning","neural-network","python","pytorch","torchviz","visualization"],"latest_commit_sha":null,"homepage":"https://mert-kurttutan.github.io/torchview/","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/mert-kurttutan.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":".github/FUNDING.yml","license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null,"zenodo":null},"funding":{"github":"mert-kurttutan"}},"created_at":"2022-11-08T22:35:11.000Z","updated_at":"2025-04-22T09:38:36.000Z","dependencies_parsed_at":"2024-06-21T07:30:06.864Z","dependency_job_id":"97a60cb3-bcd2-4b5d-b9be-e64fb79c0b08","html_url":"https://github.com/mert-kurttutan/torchview","commit_stats":{"total_commits":107,"total_committers":2,"mean_commits":53.5,"dds":0.009345794392523366,"last_synced_commit":"c640a4a55e49183fe05a8673f0a6d43943fbe84b"},"previous_names":[],"tags_count":9,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/mert-kurttutan%2Ftorchview","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/mert-kurttutan%2Ftorchview/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/mert-kurttutan%2Ftorchview/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/mert-kurttutan%2Ftorchview/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/mert-kurttutan","download_url":"https://codeload.github.com/mert-kurttutan/torchview/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":254092663,"owners_count":22013290,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["deep-learning","keras","machine-learning","neural-network","python","pytorch","torchviz","visualization"],"created_at":"2024-09-25T00:30:29.552Z","updated_at":"2025-10-23T18:37:32.586Z","avatar_url":"https://github.com/mert-kurttutan.png","language":"Python","funding_links":["https://github.com/sponsors/mert-kurttutan"],"categories":["Python"],"sub_categories":[],"readme":"# torchview\n\n[![Python 3.8+](https://img.shields.io/badge/python-3.8+-blue.svg)](https://www.python.org/downloads/release/python-380/)\n[![PyPI version](https://badge.fury.io/py/torchview.svg)](https://badge.fury.io/py/torchview)\n[![Conda version](https://img.shields.io/conda/vn/conda-forge/torchview)](https://anaconda.org/conda-forge/torchview)\n[![Build Status](https://github.com/mert-kurttutan/torchview/actions/workflows/test.yml/badge.svg)](https://github.com/mert-kurttutan/torchview/actions/workflows/test.yml)\n[![GitHub license](https://img.shields.io/github/license/mert-kurttutan/torchview)](https://github.com/mert-kurttutan/torchview/blob/main/LICENSE)\n[![codecov](https://codecov.io/gh/mert-kurttutan/torchview/branch/main/graph/badge.svg)](https://codecov.io/gh/mert-kurttutan/torchview)\n[![Downloads](https://pepy.tech/badge/torchview)](https://pepy.tech/project/torchview)\n\nTorchview provides visualization of pytorch models in the form of visual graphs. Visualization includes tensors, modules, torch.functions and info such as input/output shapes.\n\nPytorch version of `plot_model of keras` (and more)\n\nSupports PyTorch versions $\\geq$ 1.7.\n\n## Useful features\n\n\u003cp align=\"center\"\u003e\n  \u003cpicture align=\"center\"\u003e\n    \u003csource media=\"(prefers-color-scheme: dark)\" srcset=\"https://user-images.githubusercontent.com/88637659/213171745-7acf07df-6578-4a50-a106-1a7b368f8d6c.svg\"\u003e\n    \u003csource media=\"(prefers-color-scheme: light)\" srcset=\"https://user-images.githubusercontent.com/88637659/213173736-6e91724c-8de1-4568-9d52-297b4b5ff0d2.svg\"\u003e\n    \u003cimg alt=\"Shows a bar chart with benchmark results.\" src=\"https://user-images.githubusercontent.com/88637659/213173736-6e91724c-8de1-4568-9d52-297b4b5ff0d2.svg\"\u003e\n  \u003c/picture\u003e\n\u003c/p\u003e\n\n\n\n## Installation\n\nFirst, you need to install graphviz,\n\n```Bash\npip install graphviz\n```\n\nFor python interface of graphiz to work, you need to have dot layout command working in your system. If it isn't already installed, I suggest you run the following depeding on your OS,\n\nDebian-based Linux distro (e.g. Ubuntu):\n\n```Bash\napt-get install graphviz\n```\n\nWindows:\n\n```Bash\nchoco install graphviz\n```\n\nmacOS\n\n```Bash\nbrew install graphviz\n```\n\nsee more details [here](https://graphviz.readthedocs.io/en/stable/manual.html)\n\nThen, continue with installing torchview using pip\n\n```Bash\npip install torchview\n```\n\nor if you want via conda\n\n```Bash\nconda install -c conda-forge torchview\n```\n\nor if you want most up-to-date version, install directly from repo\n\n```Bash\npip install git+https://github.com/mert-kurttutan/torchview.git\n```\n\n## How To Use\n\n```python\nfrom torchview import draw_graph\n\nmodel = MLP()\nbatch_size = 2\n# device='meta' -\u003e no memory is consumed for visualization\nmodel_graph = draw_graph(model, input_size=(batch_size, 128), device='meta')\nmodel_graph.visual_graph\n```\n\n![output](https://user-images.githubusercontent.com/88637659/206028431-b114f48e-6307-4ff3-b31a-a74185eb61b5.png)\n\n## Notebook Examples\n\nFor more examples, see colab notebooks below,\n\n**Introduction Notebook:** [![Introduction](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/mert-kurttutan/torchview/blob/main/docs/docs/tutorial/notebook/example_introduction.ipynb)\n\n**Computer Vision Models:** [![Vision](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/mert-kurttutan/torchview/blob/main/docs/docs/tutorial/notebook/example_vision.ipynb)\n\n**NLP Models:** [![NLP](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/mert-kurttutan/torchview/blob/main/docs/docs/tutorial/notebook/example_text.ipynb)\n\n\u003c!-- single_input_all_cols.out --\u003e\n\n**Note:** Output graphviz visuals return images with desired sizes. But sometimes, on VSCode, some shapes are being cropped due to large size and svg rendering on by VSCode. To solve this, I suggest you run the following\n\n```python\nimport graphviz\ngraphviz.set_jupyter_format('png')\n```\n\nThis problem does not occur on other platforms e.g. JupyterLab or Google Colab.\n\n## Supported Features\n\n* Almost all the models, RNN, Sequentials, Skip Connection, Hugging Face Models\n* Support for meta tensors -\u003e no memory consumption (for very Large models) (pytorch version $\\geq$ 1.13) .\n* Shows operations between tensors (in addition to module calls)\n* Rolling/Unrolling feature. Recursively used modules can be rolled visually, see below.\n* Diverse set of inputs/output types, e.g. nested data structure (dict, list, etc), huggingface tokenizer outputs\n\n## Documentation\n\n```python\ndef draw_graph(\n    model: nn.Module,\n    input_data: INPUT_DATA_TYPE | None = None,\n    input_size: INPUT_SIZE_TYPE | None = None,\n    graph_name: str = 'model',\n    depth: int | float = 3,\n    device: torch.device | str | None = None,\n    dtypes: list[torch.dtype] | None = None,\n    mode: str | None = None,\n    strict: bool = True,\n    expand_nested: bool = False,\n    graph_dir: str | None = None,\n    hide_module_functions: bool = True,\n    hide_inner_tensors: bool = True,\n    roll: bool = False,\n    show_shapes: bool = True,\n    save_graph: bool = False,\n    filename: str | None = None,\n    directory: str = '.',\n    **kwargs: Any,\n) -\u003e ComputationGraph:\n    '''Returns visual representation of the input Pytorch Module with\n    ComputationGraph object. ComputationGraph object contains:\n\n    1) Root nodes (usually tensor node for input tensors) which connect to all\n    the other nodes of computation graph of pytorch module recorded during forward\n    propagation.\n\n    2) graphviz.Digraph object that contains visual representation of computation\n    graph of pytorch module. This graph visual shows modules/ module hierarchy,\n    torch_functions, shapes and tensors recorded during forward prop, for examples\n    see documentation, and colab notebooks.\n\n\n    Args:\n        model (nn.Module):\n            Pytorch model to represent visually.\n\n        input_data (data structure containing torch.Tensor):\n            input for forward method of model. Wrap it in a list for\n            multiple args or in a dict or kwargs\n\n        input_size (Sequence of Sizes):\n            Shape of input data as a List/Tuple/torch.Size\n            (dtypes must match model input, default is FloatTensors).\n            Default: None\n\n        graph_name (str):\n            Name for graphviz.Digraph object. Also default name graphviz file\n            of Graph Visualization\n            Default: 'model'\n\n        depth (int):\n            Upper limit for depth of nodes to be shown in visualization.\n            Depth is measured how far is module/tensor inside the module hierarchy.\n            For instance, main module has depth=0, whereas submodule of main module\n            has depth=1, and so on.\n            Default: 3\n\n        device (str or torch.device):\n            Device to place and input tensors. Defaults to\n            gpu if cuda is seen by pytorch, otherwise to cpu.\n            Default: None\n\n        dtypes (list of torch.dtype):\n            Uses dtypes to set the types of input tensor if\n            input size is given.\n\n        mode (str):\n            Mode of model to use for forward prop. Defaults\n            to Eval mode if not given\n            Default: None\n\n        strict (bool):\n            if true, graphviz visual does not allow multiple edges\n            between nodes. Mutiple edge occurs e.g. when there are tensors\n            from module node to module node and hiding those tensors\n            Default: True\n        \n        expand_nested(bool):\n            if true shows nested modules with dashed borders\n\n        graph_dir (str):\n            Sets the direction of visual graph\n            'TB' -\u003e Top to Bottom\n            'LR' -\u003e Left to Right\n            'BT' -\u003e Bottom to Top\n            'RL' -\u003e Right to Left\n            Default: None -\u003e TB\n\n        hide_module_function (bool):\n            Determines whether to hide module torch_functions. Some\n            modules consist only of torch_functions (no submodule),\n            e.g. nn.Conv2d.\n            True =\u003e Dont include module functions in graphviz\n            False =\u003e Include modules function in graphviz\n            Default: True\n\n        hide_inner_tensors (bool):\n            Inner tensor is all the tensors of computation graph\n            but input and output tensors\n            True =\u003e Does not show inner tensors in graphviz\n            False =\u003e Shows inner tensors in graphviz\n            Default: True\n\n        roll (bool):\n            If true, rolls recursive modules.\n            Default: False\n\n        show_shapes (bool):\n            True =\u003e Show shape of tensor, input, and output\n            False =\u003e Dont show\n            Default: True\n\n        save_graph (bool):\n            True =\u003e Saves output file of graphviz graph\n            False =\u003e Does not save\n            Default: False\n\n        filename (str):\n            name of the file to store dot syntax representation and\n            image file of graphviz graph. Defaults to graph_name\n\n        directory (str):\n            directory in which to store graphviz output files.\n            Default: .\n\n    Returns:\n        ComputationGraph object that contains visualization of the input\n        pytorch model in the form of graphviz Digraph object\n    '''\n```\n\n## Examples\n\n### Rolled Version of Recursive Networks\n\n```python\nfrom torchview import draw_graph\n\nmodel_graph = draw_graph(\n    SimpleRNN(), input_size=(2, 3),\n    graph_name='RecursiveNet',\n    roll=True\n)\nmodel_graph.visual_graph\n```\n\n![rnns](https://user-images.githubusercontent.com/88637659/206644016-23a89c81-1d6a-4558-82f4-33f179b345f3.png)\n\n### Show/Hide intermediate (hidden) tensors and Functionals\n\n```python\n# Show inner tensors and Functionals\nmodel_graph = draw_graph(\n    MLP(), input_size=(2, 128),\n    graph_name='MLP',\n    hide_inner_tensors=False,\n    hide_module_functions=False,\n)\n\nmodel_graph.visual_graph\n```\n\n![download](https://user-images.githubusercontent.com/88637659/206188796-4b9e57ef-8d33-469b-b8e0-2c47b06fe70b.png)\n\n\n### ResNet / Skip Connection / Support for Torch operations / Nested Modules\n\n```python\nimport torchvision\n\nmodel_graph = draw_graph(resnet18(), input_size=(1,3,32,32), expand_nested=True)\nmodel_graph.visual_graph\n```\n\n![expand_nested_resnet_model gv](https://user-images.githubusercontent.com/88637659/206036653-293f8ce7-04dd-4ac6-9de8-0061de505bba.png)\n\n## TODO\n\n* [ ] Display Module parameter info\n* [ ] Support for Graph Neural Network (GNN)\n* [ ] Support for Undirected edges for GNNs\n* [ ] Support for torch-based functions[^1]\n\n[^1]: Here, torch-based functions refers to any function that uses only torch functions and modules. This is more general than modules.\n\n## Contributing\n\nAll issues and pull requests are much appreciated! If you are wondering how to build the project:\n\n* torchview is actively developed using the latest version of Python.\n* Changes should be backward compatible to Python 3.7, and will follow Python's End-of-Life guidance for old versions.\n* Run `pip install -r requirements-dev.txt`. We use the latest versions of all dev packages.\n* To run unit tests, run `pytest`.\n* To update the expected output files, run `pytest --overwrite`.\n* To skip output file tests, use `pytest --no-output`\n\n## References\n\n* Parts related to input processing and validation are taken/inspired from torchinfo repository!!.\n* Many of the software related parts (e.g. testing) are also taken/inspired from torchinfo repository. So big thanks to @TylerYep!!!\n* The algorithm for constructing visual graphs is thanks to `__torch_function__` and subclassing `torch.Tensor`. Big thanks to all those who developed this API!!.\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fmert-kurttutan%2Ftorchview","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fmert-kurttutan%2Ftorchview","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fmert-kurttutan%2Ftorchview/lists"}