迁移学习:借助预训练模型,泛化到我们自己的模型上
举栗子VGG16网络,在ImageNet上已经训练好,是一个预训练模型,有1000个分类
1、当我们的数据集较小时,只修改最后一层全连接层,冻结其余所有层的结构和权重;
2、当我们的数据集较大时,修改所有全连接层,冻结其余所有卷积层的结构和权重;
(Transfer Learning and fine tuning)迁移学习与微调
最新推荐文章于 2024-06-11 17:55:27 发布
迁移学习:借助预训练模型,泛化到我们自己的模型上
举栗子VGG16网络,在ImageNet上已经训练好,是一个预训练模型,有1000个分类
1、当我们的数据集较小时,只修改最后一层全连接层,冻结其余所有层的结构和权重;
2、当我们的数据集较大时,修改所有全连接层,冻结其余所有卷积层的结构和权重;