使用pytorch搭建ResNeXt并基于迁移学习训练

冻结除最后全连接层以外的所有权重,只去单独训练它最后一层的的权重,这个方法,冻结了所有网络的权重。

     for param in net.parameters():
         param.requires_grad = False

使用pytorch搭建ResNeXt并基于迁移学习训练_第1张图片

你可能感兴趣的:(迁移学习,图片分类)