An open API service indexing awesome lists of open source software.

https://github.com/kentaroy47/computation-calculator.pytorch

Calculate the number of computation (MACs) you need for a given CNN.
https://github.com/kentaroy47/computation-calculator.pytorch

calculate cnn computation mac

Last synced: 7 months ago
JSON representation

Calculate the number of computation (MACs) you need for a given CNN.

Awesome Lists containing this project

README

          

# computation-calculator.pytorch
Calculate the number of computation (MACs) you need for a given CNN.

## what you need.
model.py..define your CNN model.

## run
python calculate-mac.py

## example
model.py has resnet18 inside.