Skip to content

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

Notifications You must be signed in to change notification settings

chickenjohn/CNN-Calculator

 
 

Repository files navigation

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.

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%