冻结除最后全连接层以外的所有权重,只去单独训练它最后一层的的权重,这个方法,冻结了所有网络的权重。
python
for param in net.parameters():
param.requires_grad = False
冻结除最后全连接层以外的所有权重,只去单独训练它最后一层的的权重,这个方法,冻结了所有网络的权重。
for param in net.parameters():
param.requires_grad = False