better init_params

TuringLang · Jul 3, 2023 · 4f66880 · 4f66880
1 parent 71aac85
commit 4f66880
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/contrib/inference/abstractmcmc.jl b/src/contrib/inference/abstractmcmc.jl
@@ -47,6 +47,7 @@ function AbstractMCMC.step(
     rng::Random.AbstractRNG,
     model::DynamicPPL.Model,
     sampler_wrapper::Sampler{<:ExternalSampler};
+    init_params=nothing,
     kwargs...
 )
     sampler = sampler_wrapper.alg.sampler
@@ -58,6 +59,11 @@ function AbstractMCMC.step(
     # Link the varinfo.
     f = setvarinfo(f, DynamicPPL.link!!(getvarinfo(f), model))
 
+    # Make init_params
+    if init_params === nothing
+        init_params = 4 .* rand(rng, LogDensityProblems.dimension(f)) .- 2
+    end
+
     # Then just call `AdvancedHMC.step` with the right arguments.
     transition_inner, state_inner = AbstractMCMC.step(
         rng, AbstractMCMC.LogDensityModel(f), sampler; kwargs...