def METHOD_NAME(self):
    for module in self.modules():
        if isinstance(module, nn.BatchNorm2d):
            module.eps = self.config.bn_eps
            module.momentum = self.config.bn_momentum