pytorch 迁移学习[摘自官网]

2024年4月9日 132次阅读来源: pytorch

迁移学习包含两种：微调和特征提取器。

微调：对整个网络进行训练，更新所有参数

特征提取器：只对最后的输出层训练，其他层的权重保持不变

当然，二者的共性就是需要加载训练好的权重，比如在ImageNet上训练过的vgg,resnet等等。

那么，不管是微调还是特征提取器，大致都要遵从四个步骤。

    原文作者：pytorch
    原文地址: https://www.cnblogs.com/wzyuan/p/9823340.html
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。