pytorch输出网络每个部分的梯度等

for name, param in model_stress_strain.modelensemble[0].named_parameters():
    print(name, param.grad)

输出网络每个部分的梯度。
不仅如此，还可以输出权重：

for name, param in model_stress_strain.modelensemble[0].named_parameters():
    print(name, param.data)