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版权协议,转载请附上原文出处链接和本声明。