仅作为记录,大佬请跳过。
即设置requires_grad
for name, param in model_ft.named_parameters():
print(name)
if name.startswith('fc') or name.startswith('fc2'):
param.requires_grad = True
else:
param.requires_grad = False
break
参考