分类：训练部分参数

2021
03-30

pytorch 实现冻结部分参数训练另一部分

1）添加下面一句话到模型中forpinself.parameters():p.requires_grad=False比如加载了resnet预训练模型之后，在resenet的基础上连接了新的模快，resenet模块那部分可以先暂时冻结不更新，只更新其他部分的参数，那么可以在下面加入上面那句话classRESNET_MF(nn.Module):def__init__(self,model,pretrained):super(RESNET_MF,self).__init__()self.resnet=model(pretrained)forpinself.parameters():p.re... 继续阅读 >