A simple calculator for computing the number of parameters and FLOPs in convolutional neural networks with PyTorch-like APIs.
Refer to CNNCalculator.py for details.
qinzheng93/CNN-Calculator
A simple calculator for computing the number of parameters and FLOPs in convolutional neural networks with PyTorch-like APIs.
Python