fix incorrect implicit diff H term

marius311 · Aug 27, 2023 · 4e66c41 · 4e66c41
1 parent 114123f
commit 4e66c41
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/muse.jl b/src/muse.jl
@@ -352,12 +352,12 @@ function get_H!(
                 ad_fwd, ad_rev = AD.second_lowest(prob.autodiff), AD.lowest(prob.autodiff)
 
                 ## non-implicit-diff term
-                H1 = implicit_diff_H1_is_zero ? 𝟘 : copyto!(similar(𝟘), permutedims(first(AD.jacobian(θ₀, backend=ad_fwd) do θ
+                H1 = implicit_diff_H1_is_zero ? 𝟘 : copyto!(similar(𝟘), first(AD.jacobian(θ₀, backend=ad_fwd) do θ
                     local x, = sample_x_z(prob, copy(rng), θ)
                     first(AD.gradient(θ₀, backend=ad_rev) do θ′ 
                         logLike(prob, x, ẑ, θ′, UnTransformedθ())
                     end)
-                end)))
+                end))
 
                 ## term involving dzMAP/dθ via implicit-diff (w/ conjugate-gradient linear solve)
                 dFdθ = first(AD.jacobian(θ₀, backend=ad_fwd) do θ