1.分析 *.pth的网络结构
import torch
fy=torch.load(路径)
for i in fy.keys():
print(i+' '+str(list(fy[i].size())))
2.分析自定义网络结构
import torchsummary
summary(model,input_size=(3,224,224))
产生错误input 和weight 的精度类型不匹配
import torchsummary
device = torch.device("cpu")
summary(model,device=device,input_size=(3,224,224))