pytorch迁移学习
摘要：
from torchvision import models model = models.alexnet(pretrained=True) # 提取fc层中固定的参数（这一层的输入节点数目 fc_features = model.fc.in_features # 修改类别为9，（直接对类的属性进行 阅读全文
posted @ 2021-03-09 14:36 swuxyj 阅读(66) 评论(0) 推荐(0) 编辑