flops-counter.pytorch icon indicating copy to clipboard operation
flops-counter.pytorch copied to clipboard
verified publisher icon sovrasov

the Stage of the MACs

Open niebowen666 opened this issue 1 year ago • 1 comments

The output of profile is macs and params. Does the macs just about the forward stage? Or it include the calculation complexity of forward, backward and optimization？

niebowen666 avatar Oct 30 '24 08:10 niebowen666

The estimation is done only for forward pass, backward and optimizer stages are not included

sovrasov avatar Oct 30 '24 09:10 sovrasov