RuntimeError: (NotFound) No Output(X@GRAD) found for BatchNormGrad operator. [Hint: Expected ctx-

2021/9/30 6:10:56

本文主要是介绍RuntimeError: (NotFound) No Output(X@GRAD) found for BatchNormGrad operator. [Hint: Expected ctx-,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

paddle 报错
RuntimeError: (NotFound) No Output(X@GRAD) found for BatchNormGrad operator.
[Hint: Expected ctx->HasOutput(framework::GradVarName("X")) == true, but received ctx->HasOutput(framework::GradVarName("X")):0 != true:1.] (at /paddle/paddle/fluid/operators/batch_norm_op.cc:468)

BN层的问题
修改方法:在BN层前加

if self.data_bn:
    x.stop_gradient = False
x = self.data_bn(x)


这篇关于RuntimeError: (NotFound) No Output(X@GRAD) found for BatchNormGrad operator. [Hint: Expected ctx-的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程