def METHOD_NAME(self, momentum, eps):
    for m in self.modules():
        if isinstance(m, nn.BatchNorm2d) or isinstance(m, nn.BatchNorm1d):
            m.momentum = momentum
            m.eps = eps
    return