for params in model.parameters(): params.requires_grad = False
model.layer4.requires_grad_(True) model.fc.requires_grad_(True)