Packaging to v1.0.0

.gitignore

@@ -1,3 +1,6 @@
-__pycache__
+__pycache__/
+dist/
+build/
+torchsummaryX.egg-info/
 .DS_Store
 *.swp

README.md

@@ -2,6 +2,8 @@
 Improved visualization tool of [torchsummary](https://github.com/sksq96/pytorch-summary). Here, it visualizes kernel size, output shape, # params, and Mult-Adds. Also the torchsummaryX can handle RNN, Recursive NN, or model with multiple inputs.
 
 ## Usage
+`pip install torchsummaryX` and
+
 ```python
 from torchsummaryX import summary
 summary(your_model, torch.zeros((1, 3, 224, 224)))

setup.py

@@ -0,0 +1,11 @@
+from setuptools import setup, find_packages
+
+setup(
+    name         = "torchsummaryX",
+    version      = "1.0.0",
+    description  = "Improved visualization tool of torchsummary.",
+    author       = "Namhyuk Ahn",
+    author_email = "nmhkahn@gmail.com",
+    url          = "https://github.com/nmhkahn/torchsummaryX",
+    packages     =["torchsummaryX"],
+)

torchsummaryX/__init__.py

@@ -0,0 +1 @@
+from .torchsummaryX import summary