Skip to main content

This is a tool to visualize models directly from py scripts

Project description

pytorch-show

This is a tool to visualize models directly from py scripts and it can automatically find all nn.Module in the file

RUN

python main.py -f python_file -i 240 240 -d -n img.png

image-20230627194415687

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

pytorch_show-0.1.3.tar.gz (3.7 kB view hashes)

Uploaded Source

Built Distribution

pytorch_show-0.1.3-py3-none-any.whl (4.6 kB view hashes)

Uploaded Python 3

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