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)
更多推荐
2
0- 0
已为社区贡献1条内容
所有评论(0)