flops-counter.pytorch

flops-counter.pytorch copied to clipboard

Additions and Multiplications count separately

3

Hi, This is a great repo! :-) But can you add functionality to compute the addition count and multiplication count separately? Thanks!

yashbhalgat

It would be nice to have a dictionary containing each layer/operations' GMac and Params that would be useful to plot and be able to see network bottlenecks.

carlodizon03

not suport two input image network?

9

wikiwen

# Is this repo work for all BERT model?

1

Hi, first of all thank you so much, that's a really helpful repo! I'm now working with some pretrained BERT models, so I'm wondering if this flop_counter works for all...

janezzzz

Possible redundant dataclasses dependency

1

When installing using pip with python 3.7, the following error occurs: ``` ERROR: Package 'dataclasses' requires a different Python: 3.7.9 not in '>=3.6,

AndersonZhangyq

FLOPS for LSTM model

2

How to compute FLOPS for LSTM model for skeleton data (NTU-RGBD)? Error is due to input_res parameter as for skeleton data what should be this value.

preksha12

FP16 (half percision) support?

3

I really appreciate this flops counter. Would you guys plan to support the calculation with FP16?

zhengsx

Add benchmark mode support?

3

This tool is very useful, but there is still a lack of functionality in the benchmark. 1. layer or topology mode. layer mode: enable loop for each layer to estimate...

0400H

Error when loading from a saved checkpoint file

1

Hi, I have a training script with structure as : model=Net() flops, params = get_model_complexity_info(model, (3, 256, 256), as_strings=True, print_per_layer_stat=True) print('Flops: ' + flops) print('Params: ' + params) training torch.save(model)...

PlumedSerpent

how to cite your showing two tabels

1

EtheneXiang