CNN-Calculator
CNN-Calculator copied to clipboard
A simple calculator for computing the number of parameters and FLOPs in convolutional neural networks with PyTorch-like APIs.
CNN-Calculator
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.