m0_48341098的博客

原创 解决nn.batchnorm2d报错：RuntimeError: one of the variables needed for gradient computation

这行代码可以放在train（）函数之外，或者main（）函数之外，例如定义args参数之前的parser = argparse.ArgumentParser()。这种问题往往非常头疼，因为只告诉你是梯度回传的问题，不告诉具体是哪，往往这种操作会由一些inplace操作导致。运行后发现报错位置在nn.batchnorm2d，也就是说是我的batchnorm层导致梯度无法回传。搜索了很多关于“batchnorm导致梯度无法回传”，几乎没有相关内容。，然而我检查我代码，没有此操作。随后根据报错提示，用。