首页 PaddleCV 帖子详情
无法反向传播
收藏
快速回复
PaddleCV 问答语义分割 1466 0
无法反向传播
收藏
快速回复
PaddleCV 问答语义分割 1466 0

b21=enet.block4(b20)

b22= fluid.layers.transpose(b21, perm=[0, 2, 3, 1])
out1=fluid.layers.reshape(b22,shape=[-1,9])
out2=fluid.layers.reshape(y,shape=[-1,1])
#out3=paddle.fluid.layers.softmax(out1)
_, out4 = fluid.layers.topk(out1, k=1)
out5 = fluid.layers.cast(out4, dtype='float32')
loss=paddle.fluid.layers.square_error_cost(out5, out2)
avg=fluid.layers.reduce_mean(loss)
regularizer = fluid.regularizer.L2Decay(0.0001)
optimizer = paddle.fluid.optimizer.AdamOptimizer(learning_rate=0.01, beta1=0.9, beta2=0.999, epsilon=1e-08, regularization=regularizer)
_, params_grads = optimizer.minimize(avg)

 

代码很简单就是一个9通道的图进来之后做一个语意分割反向传播,程序一直可以运行,但是就是参数不更新,如果用1张图片训练avg永远都是那个数字不变。

0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户