Batchnorm layers use running stats internally and expect the input channels to match their dimension.
The x input is using 80 channels (dim1), which raises the error and I’m unsure which workaround should be used instead.
Could you explain your idea a bit more?