Skip to main content

A tool to visualize big models

Project description

BigModelVis

version

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

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


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)

Uploaded Source

Built Distribution

bigmodelvis-0.0.1-py3-none-any.whl (13.7 kB view details)

Uploaded Python 3

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

Hashes for bigmodelvis-0.0.1.tar.gz
Algorithm Hash digest
SHA256 c5c89e41ff63389622fb13303ac9556b0bb72c338804079c96666aa629a98cfe
MD5 1a39fe5dd3ca4edc832a2b3ff953a30c
BLAKE2b-256 986dc855f6f8cb164c54736ad7d5db65a013ede7408c728f2fcb021696d9d38f

See more details on using hashes here.

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

Hashes for bigmodelvis-0.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 34ecf4b62bc07f28f023961fc52786964e3a430bd4ea9da9dda42c08960abf21
MD5 91d8d51b83ed89ffbce7beaee5e3b1cd
BLAKE2b-256 581791f21e91dc8bfbb29bc1d9c48a105875016803056022db203c6f3a07e622

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page