Can this project support count MACs of a onnx model ?

[hyuan-kamuda]

Searching "onnx MACs" to reach this project , I am finding a onnx model profiler tool like "https://github.com/Lyken17/pytorch-OpCounter" , need to use the tool to count onnx models' MACs

Thanks lot

[gmalivenko]

Hello @hyuan-kamuda.

The project is in progress now. It supports basic layers such as Conv, Linear, some activations. You can try to calculate number of MACs using CLI (see more in readme).

[hyuan-kamuda]

Hello @hyuan-kamuda.

The project is in progress now. It supports basic layers such as Conv, Linear, some activations. You can try to calculate number of MACs using CLI (see more in readme).

I tried below

My enviorment: CentOS8 Anaconda3 with Python3.7 numpy 1.19.2 onnx 1.8.1 onnxruntime 1.6.0

(py37) [hyuan@localhost ~]$conda list packages in environment at /home/hyuan/anaconda3/envs/py37:

Name Version Build Channel _libgcc_mutex 0.1 conda_forge conda-forge _openmp_mutex 4.5 1_gnu conda-forge blas 1.0 mkl defaults ca-certificates 2021.1.19 h06a4308_0 defaults certifi 2020.12.5 py37h89c1867_1 conda-forge cpuonly 1.0 0 pytorch freetype 2.10.4 h5ab3b9f_0 defaults intel-openmp 2020.2 254 defaults jpeg 9b h024ee3a_2 defaults lcms2 2.11 h396b838_0 defaults ld_impl_linux-64 2.33.1 h53a641e_7 defaults libedit 3.1.20191231 h14c3975_1 defaults libffi 3.3 he6710b0_2 defaults libgcc-ng 9.3.0 h2828fa1_18 conda-forge libgomp 9.3.0 h2828fa1_18 conda-forge libpng 1.6.37 hbc83047_0 defaults libprotobuf 3.15.2 h780b84a_0 conda-forge libstdcxx-ng 9.3.0 h6de172a_18 conda-forge libtiff 4.1.0 h2733197_1 defaults libuv 1.40.0 h7b6447c_0 defaults lz4-c 1.9.3 h2531618_0 defaults mkl 2020.2 256 defaults mkl-service 2.3.0 py37he8ac12f_0 defaults mkl_fft 1.3.0 py37h54f3939_0 defaults mkl_random 1.1.1 py37h0573a6f_0 defaults ncurses 6.2 he6710b0_1 defaults ninja 1.10.2 py37hff7bd54_0 defaults numpy 1.19.2 py37h54aff64_0 defaults numpy-base 1.19.2 py37hfa32c7d_0 defaults olefile 0.46 py37_0 defaults onnx 1.8.1 py37h7d3e3ad_1 conda-forge onnxruntime 1.6.0 pypi_0 pypi openssl 1.1.1j h27cfd23_0 defaults pillow 8.1.0 py37he98fc37_0 defaults pip 21.0.1 py37h06a4308_0 defaults protobuf 3.15.2 py37hcd2ae1e_0 conda-forge python 3.7.10 hdb3f193_0 defaults python_abi 3.7 1_cp37m conda-forge pytorch 1.7.1 py3.7_cpu_0 [cpuonly] pytorch readline 8.1 h27cfd23_0 defaults setuptools 52.0.0 py37h06a4308_0 defaults six 1.15.0 py37h06a4308_0 defaults sqlite 3.33.0 h62c20be_0 defaults tk 8.6.10 hbc83047_0 defaults torchaudio 0.7.2 py37 pytorch torchvision 0.8.2 py37_cpu [cpuonly] pytorch typing-extensions 3.7.4.3 0 conda-forge typing_extensions 3.7.4.3 py_0 conda-forge wheel 0.36.2 pyhd3eb1b0_0 defaults xz 5.2.5 h7b6447c_0 defaults zlib 1.2.11 h7b6447c_3 defaults zstd 1.4.5 h9ceee32_0 defaults

Following your project readme install but met below errors, could you pls help?

$ pip install onnx_opcounter Collecting onnx_opcounter Using cached onnx_opcounter-0.0.2.tar.gz (3.3 kB) ERROR: Command errored out with exit status 1: command: /home/hyuan/anaconda3/envs/py37/bin/python -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-mv0no8x3/onnx-opcounter_610c3423f4e54dbbab28f1241d9522c3/setup.py'"'"'; file='"'"'/tmp/pip-install-mv0no8x3/onnx-opcounter_610c3423f4e54dbbab28f1241d9522c3/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(file);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, file, '"'"'exec'"'"'))' egg_info --egg-base /tmp/pip-pip-egg-info-zop4ylid cwd: /tmp/pip-install-mv0no8x3/onnx-opcounter_610c3423f4e54dbbab28f1241d9522c3/ Complete output (7 lines): Traceback (most recent call last): File "", line 1, in File "/tmp/pip-install-mv0no8x3/onnx-opcounter_610c3423f4e54dbbab28f1241d9522c3/setup.py", line 9, in reqs = parse_requirements('requirements.txt') File "/tmp/pip-install-mv0no8x3/onnx-opcounter_610c3423f4e54dbbab28f1241d9522c3/setup.py", line 5, in parse_requirements lineiter = (line.strip() for line in open(filename)) FileNotFoundError: [Errno 2] No such file or directory: 'requirements.txt'

WARNING: Discarding https://files.pythonhosted.org/packages/aa/19/f202a25f37822819ad12cf04e9be3447ad781b50f45e17e32be4221a165f/onnx_opcounter-0.0.2.tar.gz#sha256=66318aa30ce5f529f15cc80e33b541bc275f7df4043caa47679650589ee58978 (from https://pypi.org/simple/onnx-opcounter/). Command errored out with exit status 1: python setup.py egg_info Check the logs for full command output. Downloading onnx_opcounter-0.0.1.tar.gz (2.6 kB) ERROR: Command errored out with exit status 1: command: /home/hyuan/anaconda3/envs/py37/bin/python -c 'import sys, setuptools, tokenize; sys.argv[0] = '"'"'/tmp/pip-install-mv0no8x3/onnx-opcounter_3b2f90e65da64ac98910c4fccedd63df/setup.py'"'"'; file='"'"'/tmp/pip-install-mv0no8x3/onnx-opcounter_3b2f90e65da64ac98910c4fccedd63df/setup.py'"'"';f=getattr(tokenize, '"'"'open'"'"', open)(file);code=f.read().replace('"'"'\r\n'"'"', '"'"'\n'"'"');f.close();exec(compile(code, file, '"'"'exec'"'"'))' egg_info --egg-base /tmp/pip-pip-egg-info-azu54_5y cwd: /tmp/pip-install-mv0no8x3/onnx-opcounter_3b2f90e65da64ac98910c4fccedd63df/ Complete output (7 lines): Traceback (most recent call last): File "", line 1, in File "/tmp/pip-install-mv0no8x3/onnx-opcounter_3b2f90e65da64ac98910c4fccedd63df/setup.py", line 9, in reqs = parse_requirements('requirements.txt') File "/tmp/pip-install-mv0no8x3/onnx-opcounter_3b2f90e65da64ac98910c4fccedd63df/setup.py", line 5, in parse_requirements lineiter = (line.strip() for line in open(filename)) FileNotFoundError: [Errno 2] No such file or directory: 'requirements.txt'

WARNING: Discarding https://files.pythonhosted.org/packages/41/9c/ed84291bfbc19f791462677395f2466654d58c26497146db7446dc6577d3/onnx_opcounter-0.0.1.tar.gz#sha256=903b54b7f7c1892173f6db5017a2d2fa87da8d3f2546599ce9e15e029e02c1d3 (from https://pypi.org/simple/onnx-opcounter/). Command errored out with exit status 1: python setup.py egg_info Check the logs for full command output. ERROR: Could not find a version that satisfies the requirement onnx_opcounter ERROR: No matching distribution found for onnx_opcounter