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
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.0.tar.gz
(3.2 kB
view hashes)
Built Distribution
mxop-0.1.0-py3-none-any.whl
(4.7 kB
view hashes)