A tool to count OPs and paramters of MXNet model.
Project description
MXOP: MXNet-OpSummary
It only works for gluon yet.
Reference: THOP: PyTorch-OpCounter
Installation
- PyPi
pip install --index-url https://pypi.org/simple/ mxop
- Github (latest)
pip install --upgrade git+https://github.com/hey-yahei/OpSummary.MXNet.git
Usage
Gluon
- Count OPs
from mxop.gluon import count_ops op_counter = count_ops(net) # net is the gluon model you want to count OPs
- Count parameters
from mxop.gluon import count_params op_counter = count_params(net, input_size) # net is the gluon model you want to count OPs # input_size is the shape of your input
- Print summary
from mxop.gluon import op_summary op_counter = op_summary(net, input_size) # net is the gluon model you want to count OPs # input_size is the shape of your input
TODO
- Count OPs and parameters for each layer.
- Support Symbol model for MXNet.
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
mxop-0.1.2.tar.gz
(3.3 kB
view hashes)
Built Distribution
mxop-0.1.2-py3-none-any.whl
(4.8 kB
view hashes)