MXnet输出网络模型权重和偏置的名称，值，梯度

net = Mynet()
net.initialize()
with ag.record():
y_pred = net(x)
y_pred.backward()

for n,v in zip(net.collect_params(),net.collect_params().values()):
print(n+' grad',v.grad())#输出关于w和b的梯度值   名称
print(n+' data',v.data())#输出w和b的值  名称

• 点赞
• 收藏
• 分享
• 文章举报