make find_good_eps friendly to Float32

TuringLang · Aug 15, 2019 · 17bbff6 · 17bbff6
1 parent db754a2
commit 17bbff6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/trajectory.jl b/src/trajectory.jl
@@ -425,9 +425,9 @@ function find_good_eps(
     max_n_iters::Int=100
 ) where {T<:Real}
     # Initialize searching parameters
-    ϵ′ = ϵ = 0.1
-    a_min, a_cross, a_max = 0.25, 0.5, 0.75 # minimal, crossing, maximal accept ratio
-    d = 2.0
+    ϵ′ = ϵ = T(0.1)
+    a_min, a_cross, a_max = T(0.25), T(0.5), T(0.75) # minimal, crossing, maximal accept ratio
+    d = T(2.0)
     # Create starting phase point
     r = rand(rng, h.metric) # sample momentum variable
     z = phasepoint(h, θ, r)