2021-09-23

一、如何计算参数量和计算量

import torch
import torchvision.models as models
# import torch
from ptflops import get_model_complexity_info

model = resnet34(num_classes=5)
model_weight_path ='xxxx'
model.load_state_dict(torch.load(model_weight_path),False)
model_name = 'xxxx'
flops, params = get_model_complexity_info(model, (3, 300, 300), as_strings=True, print_per_layer_stat=True)
# print("%s |%s |%s" % (model_name, flops, params))
print("model_name",model_name)
print("flops:",flops)
print("params:",params)
————————————————
版权声明：本文为CSDN博主「dear_queen」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/dear_queen/article/details/115579980