冻结所有参数:
def freeze_whole_model(self):
for n, p in self.model.named_parameters():
p.requires_grad = False
冻结所有参数:
def freeze_whole_model(self):
for n, p in self.model.named_parameters():
p.requires_grad = False