Hello! I’m trying to work with NNVM’s CorrectLayout pass for NHWC format. The computation graph is intialized to NCHW format. It looks like if I pass layout=‘NHWC’ to the build() function in the nnvm/python/nnvm/compiler/build_module.py, the CorrectLayout pass doesn’t change the node’s parameters, e.g. BatchNorm’s ‘axis’ parameter, which is set to 1 as default for NCHW, and therefore it doesn’t give the correct layouts for nodes in a graph.
How can I change a node’s parameters if necessary? Thanks in advance!