Awesome Lists

pytorch-OpCounter

by Lyken17

Pythonpushed over 2 years ago

Count the MACs / FLOPs of your PyTorch model.

AI summary

Model profiler

An open-source tool for counting the number of multiplication operations (MACs) and floating-point operations (FLOPs) in PyTorch models.

stars
4.9K
forks
528
watching
29

Add a GitHub project

Missing a project or an awesome list? Paste its GitHub URL and we fetch it right away.