Find step_size if zero (#348)

* Find step_size if zero * Update abstractmcmc.jl * Apply suggestions from code review Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> * Update Project.toml * Update abstractmcmc.jl * Apply suggestions from code review Co-authored-by: Tor Erlend Fjelde <tor.erlend95@gmail.com> * reuse some code. --------- Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> Co-authored-by: Tor Erlend Fjelde <tor.erlend95@gmail.com>
TuringLang · Sep 4, 2023 · 1dff529 · 1dff529 · yebai · Sep 4, 2023
1 parent 3b6351f
commit 1dff529
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "AdvancedHMC"
 uuid = "0bf59076-c3b1-5ca4-86bd-e02cd72cde3d"
-version = "0.5.4"
+version = "0.5.5"
 
 [deps]
 AbstractMCMC = "80f14c24-f653-4e6a-9b94-39d6b0f70001"

diff --git a/src/abstractmcmc.jl b/src/abstractmcmc.jl
@@ -273,7 +273,9 @@ function make_step_size(
     hamiltonian::Hamiltonian,
     init_params,
 )
-    return spl.κ.τ.integrator.ϵ
+    T = typeof(spl.κ.τ.integrator.ϵ)
+    ϵ = make_step_size(rng, spl.κ.τ.integrator, T, hamiltonian, init_params)
+    return ϵ
 end
 
 function make_step_size(
@@ -294,7 +296,13 @@ function make_step_size(
     hamiltonian::Hamiltonian,
     init_params,
 )
-    return integrator.ϵ
+    if integrator.ϵ > 0
+        ϵ = integrator.ϵ
+    else
+        ϵ = find_good_stepsize(rng, hamiltonian, init_params)
+        @info string("Found initial step size ", ϵ)
+    end
+    return T(ϵ)
 end
 
 function make_step_size(