Skip to content

Commit 3c5b56f

Browse files
committed
fix include SubSampleObjective as part of ParamSpaceSGD
1 parent e49c680 commit 3c5b56f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/algorithms/paramspacesgd/interface.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
11

22
"""
3-
This family of algorithms (`<:KLMinRepGradDescent`,`<:KLMinRepGradProxDescent`,`<:KLMinScoreGradDescent`) applies stochastic gradient descent (SGD) to the variational `objective` over the (Euclidean) space of variational parameters.
3+
This family of algorithms (`<:KLMinRepGradDescent`,`<:KLMinRepGradProxDescent`,`<:KLMinScoreGradDescent`,`<:SubsampledObjective`) applies stochastic gradient descent (SGD) to the variational `objective` over the (Euclidean) space of variational parameters.
44
The trainable parameters in the variational approximation are expected to be extractable through `Optimisers.destructure`.
55
This requires the variational approximation to be marked as a functor through `Functors.@functor`.
66
"""
77
const ParamSpaceSGD = Union{
8-
<:KLMinRepGradDescent,<:KLMinRepGradProxDescent,<:KLMinScoreGradDescent
8+
<:KLMinRepGradDescent,<:KLMinRepGradProxDescent,<:KLMinScoreGradDescent,<:SubsampledObjective
99
}
1010

1111
function init(rng::Random.AbstractRNG, alg::ParamSpaceSGD, q_init, prob)

0 commit comments

Comments
 (0)