Method __init__

diff2flow/openai_diffusion/timestep_sampler.py:63–65

(self, diffusion)

class UniformSampler(ScheduleSampler):
    def __init__(self, diffusion):
        self.diffusion = diffusion
        self._weights = np.ones([diffusion.num_timesteps])

    def weights(self):
        return self._weights