Method __init__

schedule/schedule.py:6–8

(self, schedule, timesteps)

class Schedule:
    def __init__(self, schedule, timesteps):
        self.timesteps = timesteps
        self.schedule = schedule

    def cosine_beta_schedule(self, s=0.001):
        timesteps = self.timesteps