Method limiter

diffbir/sampler/k_diffusion.py:320–321

(self, x)

        self.errs = []

    def limiter(self, x):
        return 1 + math.atan(x - 1)

    def propose_step(self, error):
        inv_error = 1 / (float(error) + self.eps)