Spaces:
Runtime error
Runtime error
File size: 695 Bytes
83d8d3c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
import argparse
from backbones import get_model
from ptflops import get_model_complexity_info
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))
|