Spaces:
Build error
Build error
File size: 686 Bytes
8eb4303 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from ptflops import get_model_complexity_info
from backbones import get_model
import argparse
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='')
parser.add_argument('n', type=str, default="r100")
args = parser.parse_args()
net = get_model(args.n)
macs, params = get_model_complexity_info(
net, (3, 112, 112), as_strings=False,
print_per_layer_stat=True, verbose=True)
gmacs = macs / (1000**3)
print("%.3f GFLOPs"%gmacs)
print("%.3f Mparams"%(params/(1000**2)))
if hasattr(net, "extra_gflops"):
print("%.3f Extra-GFLOPs"%net.extra_gflops)
print("%.3f Total-GFLOPs"%(gmacs+net.extra_gflops))
|