You might need to install timm!
pip install timm
Simply run
python measure_latency.py --model_name EffNetV2-b0
python measure_latency.py --model_name EffNetV2-b1
python measure_latency.py --model_name RevCol-T
python measure_latency.py --model_name MBNetV2
python measure_latency.py --model_name MBNetV3
python measure_latency.py --model_name MBNetV3_075
python measure_latency.py --model_name LPIENet-XXS
python measure_latency.py --model_name LPIENet-XS
python measure_latency.py --model_name LPIENet
python measure_latency.py --model_name LPIENet-L