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的值  名称

版权声明:本文为qq_42358080原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。