美文网首页
Transfer learning using Pytorch

Transfer learning using Pytorch

作者: 十年磨剑_莫回首 | 来源:发表于2021-11-25 11:33 被阅读0次

    迁移学习的两种基本应用场景:

    1 模型初始化又叫做 finetuning, 一般的模型初始化是一个随机初始,但是使用训练好的参数作为初始化,可能模型会更快的收敛和有更好的表现性能。

    2 将训练好的模型及其参数固定,在该模型后面增加新的网络或者在训练好的模型中间,替换新的网络,以期达到一种固定预定义的特征选择器作用。


    1 模型初始化

    加载预训练模型,然后将最后的全连接层的输出层替换成新的全连接,使输出有2类,当替换新的连接层以后,我们开始新的训练任务。

    2 固定的特征提取器

    这里我们需要冻结网络的梯度更新,除了最后的全连接层,通过设置特征提取网络的requires_grad=False 来冻结训练好的参数,这样他们在loss.backward中,就不会计算梯度并更新参数了。当我们替换掉原始网络中的某一层网络时,新的这一层网络的requires_grad=True, 所以在训练和loss.backward中会计算梯度并更新参数。

    相关文章

      网友评论

          本文标题:Transfer learning using Pytorch

          本文链接:https://www.haomeiwen.com/subject/cxpatrtx.html