Metadata-Version: 2.1
Name: bigmodelvis
Version: 0.0.1
Summary: A tool to visualize big models
Home-page: https://github.com/zt-wang19/bigmodelvis
Author: 
Author-email: shengdinghu@gmail.com
License: Apache
Keywords: transformers,Visualizaiton,AI,DeepLearning
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Education
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.6.0
Description-Content-Type: text/markdown
License-File: LICENSE

# BigModelVis

![version](https://img.shields.io/badge/version-0.0.1-blue)

A Tool to Visualize Big Models.

## Overview

You can use this tool to visualize the tree structure of big models, with the name/shape of each layer showing in indents.

- A demo picture of using BigModelVis to show the structure of Bert (bert-base-uncased)

![bert_vis](./img/bert_vis.png)

## Installation

install the latest pip version

```
pip install bigmodelvis
```

**or** build from source

```
[TODO] add the git address
git clone xxx
cd bigmodelvis
python setup.py install
```



## Usage

```
from bigmodelvis import Visualization
from transformers import BertForMaskedLM
model = BertForMaskedLM.from_pretrained('bert-base-uncased')
visobj = Visualization(model)
visobj.structure_graph()
```

