mirror of
https://github.com/Laurent2916/Diffusers.jl.git
synced 2024-10-17 23:26:19 +00:00
🧪 (BetaSchedules) add variance schedules monotonicity check
This commit is contained in:
parent
8c46a27423
commit
8dd1a71268
18
test/BetaSchedules.jl
Normal file
18
test/BetaSchedules.jl
Normal file
|
@ -0,0 +1,18 @@
|
|||
using Diffusers.BetaSchedules
|
||||
using Test
|
||||
|
||||
@testset "Variance schedules tests" begin
|
||||
@testset "β increases monotonically" begin
|
||||
T = 1000
|
||||
|
||||
β_linear = linear_beta_schedule(T)
|
||||
β_scaled_linear = scaled_linear_beta_schedule(T)
|
||||
β_cosine = cosine_beta_schedule(T)
|
||||
β_sigmoid = sigmoid_beta_schedule(T)
|
||||
|
||||
@test all(diff(β_linear) .>= 0)
|
||||
@test all(diff(β_scaled_linear) .>= 0)
|
||||
@test all(diff(β_cosine) .>= 0)
|
||||
@test all(diff(β_sigmoid) .>= 0)
|
||||
end
|
||||
end
|
|
@ -2,3 +2,5 @@ using Diffusers
|
|||
using Aqua
|
||||
|
||||
Aqua.test_all(Diffusers)
|
||||
|
||||
include("BetaSchedules.jl")
|
||||
|
|
Loading…
Reference in a new issue