Skip to content

Commit d6fbc22

Browse files
committed
removed FT for disambiguation
1 parent bd6aa41 commit d6fbc22

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/Accelerators.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ end
3131
"""
3232
Sets u_prev to the initial parameter values
3333
"""
34-
function set_ICs!(accelerator::NesterovAccelerator{FT}, u::MA) where {FT <: AbstractFloat, MA <: AbstractMatrix{FT}}
34+
function set_ICs!(accelerator::NesterovAccelerator{FT}, u::MA) where {FT <: AbstractFloat, MA <: AbstractMatrix}
3535
accelerator.u_prev = u
3636
end
3737

@@ -42,7 +42,7 @@ Performs traditional state update with no momentum.
4242
function accelerate!(
4343
ekp::EnsembleKalmanProcess{FT, IT, P, LRS, DefaultAccelerator},
4444
u::MA,
45-
) where {FT <: AbstractFloat, IT <: Int, P <: Process, LRS <: LearningRateScheduler, MA <: AbstractMatrix{FT}}
45+
) where {FT <: AbstractFloat, IT <: Int, P <: Process, LRS <: LearningRateScheduler, MA <: AbstractMatrix}
4646
push!(ekp.u, DataContainer(u, data_are_columns = true))
4747
end
4848

@@ -52,7 +52,7 @@ Performs state update with modified Nesterov momentum approach.
5252
function accelerate!(
5353
ekp::EnsembleKalmanProcess{FT, IT, P, LRS, NesterovAccelerator{FT}},
5454
u::MA,
55-
) where {FT <: AbstractFloat, IT <: Int, P <: Process, LRS <: LearningRateScheduler, MA <: AbstractMatrix{FT}}
55+
) where {FT <: AbstractFloat, IT <: Int, P <: Process, LRS <: LearningRateScheduler, MA <: AbstractMatrix}
5656
## update "v" state:
5757
k = get_N_iterations(ekp) + 2
5858
v = u .+ (1 - ekp.accelerator.r / k) * (u .- ekp.accelerator.u_prev)

0 commit comments

Comments
 (0)