参数固定
self.teacher_model.eval()
with torch.no_grad():
outputs = self.teacher_model.module.teacher_forward(data)
查看Conv层
conv.weight
conv.weight.requires_grad
查看BN层
bn.running_var
bn.running_mean
bn.running_var.requires_grad
bn.running_mean.requires_grad