/CNN-Calculator

A simple calculator for computing the number of parameters and FLOPs in convolutional neural networks with PyTorch-like APIs.

Primary LanguagePython

Stargazers