如何实现AdaBN?

AdaBN,主要解决训练集和测试集分布不一致问题。首先,用训练集训练一个含有BN层的网络,直至训练完成。然后,将测试集输入到训练好的模型中,只进行数据正向传播,将所有BN层的均值方差用测试集的均值方差替换,其他网络参数保持不变。
请问具体代码应该怎样实现?

很简单啊,前向计算测试集的时候,先遍历模型参数,替换BN层均值和方差。没讲有人这么做啊,你这是哪篇论文?