PreNAS icon indicating copy to clipboard operation
PreNAS copied to clipboard

Question regarding flops calculation

Open maryanpetruk opened this issue 2 years ago • 3 comments

Dear authors, @xiuyu-sxy

I would like to verify the performance measures you have provided in your work, and particularly flops calculation. Could you please let me know how one can reproduce flops numbers from the table in the Readme? image

Thank you

maryanpetruk avatar Aug 18 '23 13:08 maryanpetruk

You can use the get_complexity function in Vision_TransformerSuper to obtain the flops numbers.

BeachWang avatar Aug 21 '23 06:08 BeachWang

You can use the get_complexity function in Vision_TransformerSuper to obtain the flops numbers.

Sure, but I don't understand what the sequence_length should be to get the numbers you have. It is inconsistent for three supernets: tiny, small and base. Could you be so kind as to provide the value for the sequence_length parameter or values if those are different for the three supernets?

maryanpetruk avatar Aug 21 '23 07:08 maryanpetruk

The image size is 224x224 in ImageNet. The patch size in ViT is 16x16. So I think the sequence_length is 14x14 for three supernets.

BeachWang avatar Aug 23 '23 12:08 BeachWang