├── .gitignore ├── README.md ├── example.py ├── image └── profiling.png ├── model ├── __init__.py └── alexnet.py └── profiling.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | *.sublime* 3 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhuwenxi/pytorch-profiling-tool/HEAD/README.md -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhuwenxi/pytorch-profiling-tool/HEAD/example.py -------------------------------------------------------------------------------- /image/profiling.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhuwenxi/pytorch-profiling-tool/HEAD/image/profiling.png -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhuwenxi/pytorch-profiling-tool/HEAD/model/alexnet.py -------------------------------------------------------------------------------- /profiling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhuwenxi/pytorch-profiling-tool/HEAD/profiling.py --------------------------------------------------------------------------------