onnx-opcounter
onnx-opcounter copied to clipboard
Count number of parameters / MACs / FLOPS for ONNX models.
(py36) $ onnx_opcounter --calculate-macs dd3d_b1_cv22.onnx Number of parameters in the model: 6985714 Conv_0 -> Conv_0 index 5 Relu_2 -> Relu_2 index 6 MaxPool_3 -> MaxPool_3 index 7 Conv_4 -> Conv_4...
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...
Hello I inferenced the model Faster R-CNN from model Zoo ONNX and an error occurs during the `caclulate_macs`: ``` raise ValueError( ValueError: This ORT build has ['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider'] enabled....
Command used onnx_onnx_opcounter --calculate-macs Traceback (most recent call last): File "/env/bin/onnx_opcounter", line 8, in sys.exit(main()) File "/env/lib/python3.8/site-packages/onnx_opcounter/cli.py", line 17, in main print('Number of MACs in the model: {}'.format(calculate_macs(model))) File "/env/lib/python3.8/site-packages/onnx_opcounter/onnx_opcounter.py",...