Method get_betas

schedule/schedule.py:41–47

(self)

        return torch.sigmoid(betas) * (beta_end - beta_start) + beta_start

    def get_betas(self):
        if self.schedule == "linear":
            return self.linear_beta_schedule()
        elif self.schedule == 'cosine':
            return self.cosine_beta_schedule()
        else:
            raise NotImplementedError


if __name__ == "__main__":