Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Are MACs values equal across devices (GPU/CPU)?

wnntju opened this issue · comments

The authors' examples of input and model are both in CPU form. I want to make sure that the value of MACs is not related to GPU or CPU.

MACs is a metric that reflects how many muplications a model has performed. This is platform-agnostic, which should be same across hardware.