A tool to visualize big models
Project description
BigModelVis
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)
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()
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
bigmodelvis-0.0.1.tar.gz
(13.6 kB
view details)
Built Distribution
File details
Details for the file bigmodelvis-0.0.1.tar.gz
.
File metadata
- Download URL: bigmodelvis-0.0.1.tar.gz
- Upload date:
- Size: 13.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.9.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | c5c89e41ff63389622fb13303ac9556b0bb72c338804079c96666aa629a98cfe |
|
MD5 | 1a39fe5dd3ca4edc832a2b3ff953a30c |
|
BLAKE2b-256 | 986dc855f6f8cb164c54736ad7d5db65a013ede7408c728f2fcb021696d9d38f |
Provenance
File details
Details for the file bigmodelvis-0.0.1-py3-none-any.whl
.
File metadata
- Download URL: bigmodelvis-0.0.1-py3-none-any.whl
- Upload date:
- Size: 13.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.9.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 34ecf4b62bc07f28f023961fc52786964e3a430bd4ea9da9dda42c08960abf21 |
|
MD5 | 91d8d51b83ed89ffbce7beaee5e3b1cd |
|
BLAKE2b-256 | 581791f21e91dc8bfbb29bc1d9c48a105875016803056022db203c6f3a07e622 |