赞
踩
import torchvision
from torchvision import models
resnet50 = models.resnet50(pretrained=True) #pretrained=True 加载模型以及训练过的参数
print(resnet50) # 打印输出观察一下resnet50到底是怎么样的结构
resnet50=models.resnet50(pretrained=True)
num_in_ftrs=resnet.fc.in_features
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。