taobiaoli/onnx_opcalculator

Counts the layers number of parameters / MACs for ONNX models and save csv format.

ONNX Operations Counter

Counts the layers number of parameters / MACs for ONNX models and save csv format.

Using

python onnx_opcalculator.py --calculate-macs {path_to_onnx_model}

License

The software is covered by Apache License 2.0.

Reference

[onnx_opcounter](gmalivenko/onnx-opcounter: Count number of parameters / MACs / FLOPS for ONNX models. (github.com))