Metadata-Version: 2.1
Name: pytorch-show
Version: 0.1.2
Summary: This is a tool to visualize models directly from py scripts
Author: Yshelgi
Author-email: 3041696596@qq.com
Requires-Python: >=3.8,<4.0
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Requires-Dist: tabulate (>=0.9.0,<0.10.0)
Requires-Dist: torch (>=2.0.0,<3.0.0)
Requires-Dist: torchvision (>=0.15.1,<0.16.0)
Description-Content-Type: text/markdown

# 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](https://s2.loli.net/2023/06/27/k4psbUREQHlXV2O.png)

![image-20230627194415687](https://s2.loli.net/2023/06/27/ahbLlf6yOZ57KYw.png)
