AI实战:pytorch版ResNet迁移学习前言ResNet详解加载预训练模型构建base model,作为特征层环境代码支持的模型数据用法参考
前言我的另外一篇基于tensorflow的迁移学习文章:AI实战:迁移学习之使用ResNet做分类本文介绍pytorch版ResNet迁移学习。本质:迁移学习到的是特征。加载预训练模型特征层 model.features分类层 model.classifier构建base model,作为特征层model = models.densenet18(pretrained=True)其他的网络可以参考: https://pytorch.org/docs/0.3.0/torchvisio