Created by: Xreki
图像分类模型中,当设置data_format=“NHWC”时,图像数据依旧按NCHW读进来,之后通过一个tranpose转换成NHWC格式。feed数据都是设置stop_gradient=True,直接传给conv2d,则该conv2d_grad不需要计算input_grad。插入transpose后,feed数据的stop_gradient属性没有传递给tranpose的输出变量,导致了conv2d_grad中产生了多余的计算(计算input_grad)。
这个PR将feed image的stop_gradient属性值传递给transpose的输出变量。