Lyken17 / pytorch-OpCounter

Count the MACs / FLOPs of your PyTorch model.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Res add will be included when evaluate Resnet's OPs ?

hollyaoyaozi opened this issue · comments

Hi, deer auther,

I wonder if thop will count residual connection's OPs (Add OP) when profiling resnet ?

Thanks

@hollyaoyaozi Try to use torch-operation-counter since it wrap the native pytorch operations (including add operation).