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 = '
原创
2021-09-23 20:55:59 ·
229 阅读 ·
0 评论