How to get the FLOPs of a model

google-research / nasbench

NASBench: A Neural Architecture Search Dataset and Benchmark

Apache License 2.0

682 stars 129 forks source link

Open zheng-ningxin opened 4 years ago

zheng-ningxin commented 4 years ago

Hi~ Is there any way to get the FLOPs of an existing model in the search space? Thanks~

AlbertiPot commented 3 years ago

Yes, the FLOPs are vital properties for DNN. Is there any function to help to calculate the subnets` FLOPs

ngctnnnn commented 2 years ago

In NasBench201, the authors have proposed this tool to measure the FLOPs from a constructed model that is found in the Bench. https://github.com/D-X-Y/AutoDL-Projects/blob/d04edcd211604e3bb3e682ba5606c0c0c5e991ce/xautodl/utils/flop_benchmark.py