semvec/pypi-packages huggingface dataset:
bigmodelvis 0.0.1
A tool to visualize big models
License: Apache
Requires Python: >=3.6.0
Keywords: transformers,Visualizaiton,AI,DeepLearning

# 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()
```