Metadata-Version: 2.1
Name: mxop
Version: 0.1.1
Summary: A tool to count OPs and paramters of MXNet model.
Home-page: https://github.com/hey-yahei/OpSummary.MXNet
Author: YaHei
Author-email: hey-yahei@qq.com
License: MIT
Description: ## MXOP: MXNet-OpSummary    
        It only works for **gluon** yet.     
            
        Reference: [THOP: PyTorch-OpCounter](https://github.com/Lyken17/pytorch-OpCounter)    
        
        ### Installation    
        * PyPi    
            ```bash
            pip install --index-url https://pypi.org/simple/ mxop
            ```
        * Github (latest)    
            ```bash
            pip install --upgrade git+https://github.com/hey-yahei/OpSummary.MXNet.git
            ```
        
        ### Usage
        #### Gluon
        * Count OPs    
            ```python
            from mxop.gluon import count_ops
            op_counter = count_ops(net)   # net is the gluon model you want to count OPs 
            ```
        * Count parameters    
            ```python
            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     
            ```python
            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.      
         
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Description-Content-Type: text/markdown
