参考 param.requires_grad = False的作用 - 云+社区 - 腾讯云
param.requires_grad = False的作用是:
屏蔽预训练模型的权重。
只训练最后一层的全连接的权重。
最后一层的添加,是通过[2]中的一句代码:
model_conv.fc = nn.Linear(num_ftrs, 2)
可以发现,冻结预训练模型的权重后,验证精度会下降。
参考 param.requires_grad = False的作用 - 云+社区 - 腾讯云
param.requires_grad = False的作用是:
屏蔽预训练模型的权重。
只训练最后一层的全连接的权重。
最后一层的添加,是通过[2]中的一句代码:
model_conv.fc = nn.Linear(num_ftrs, 2)
可以发现,冻结预训练模型的权重后,验证精度会下降。