From e029d088d0656e89f86d8d3601e4cbc70150bc02 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Laure=CE=B7t?= Date: Sun, 3 Sep 2023 13:58:01 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=A7=AA=20(BetaSchedules)=20add=20Exponent?= =?UTF-8?q?ial=20schedule=20SNR=20test?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- test/BetaSchedules.jl | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/BetaSchedules.jl b/test/BetaSchedules.jl index a186e05..b90ad4c 100644 --- a/test/BetaSchedules.jl +++ b/test/BetaSchedules.jl @@ -9,21 +9,25 @@ using Test β_scaled_linear = scaled_linear_beta_schedule(T) β_cosine = cosine_beta_schedule(T) β_sigmoid = sigmoid_beta_schedule(T) + β_exponential = exponential_beta_schedule(T) α̅_linear = cumprod(1 .- β_linear) α̅_scaled_linear = cumprod(1 .- β_scaled_linear) α̅_cosine = cumprod(1 .- β_cosine) α̅_sigmoid = cumprod(1 .- β_sigmoid) + α̅_exponential = cumprod(1 .- β_exponential) # arxiv:2208.11970 Eq. 109 SNR_linear = α̅_linear ./ (1 .- α̅_linear) SNR_scaled_linear = α̅_scaled_linear ./ (1 .- α̅_scaled_linear) SNR_cosine = α̅_cosine ./ (1 .- α̅_cosine) SNR_sigmoid = α̅_sigmoid ./ (1 .- α̅_sigmoid) + SNR_exponential = α̅_exponential ./ (1 .- α̅_exponential) @test all(diff(SNR_linear) .<= 0) @test all(diff(SNR_scaled_linear) .<= 0) @test all(diff(SNR_cosine) .<= 0) @test all(diff(SNR_sigmoid) .<= 0) + @test all(diff(SNR_exponential) .<= 0) end end