TuringLang
diff --git a/‎src/AdvancedVI.jl‎
Lines changed: 0 additions & 1 deletion b/‎src/AdvancedVI.jl‎
Lines changed: 0 additions & 1 deletion
diff --git a/‎…/algorithms/paramspacesgd/repgradelbo.jl‎ ‎test/algorithms/repgradelbo.jl‎test/algorithms/paramspacesgd/repgradelbo.jl renamed to test/algorithms/repgradelbo.jl b/‎…/algorithms/paramspacesgd/repgradelbo.jl‎ ‎test/algorithms/repgradelbo.jl‎test/algorithms/paramspacesgd/repgradelbo.jl renamed to test/algorithms/repgradelbo.jl
diff --git a/‎…ramspacesgd/repgradelbo_locationscale.jl‎ ‎…/algorithms/repgradelbo_locationscale.jl‎test/algorithms/paramspacesgd/repgradelbo_locationscale.jl renamed to test/algorithms/repgradelbo_locationscale.jl b/‎…ramspacesgd/repgradelbo_locationscale.jl‎ ‎…/algorithms/repgradelbo_locationscale.jl‎test/algorithms/paramspacesgd/repgradelbo_locationscale.jl renamed to test/algorithms/repgradelbo_locationscale.jl
diff --git a/‎…d/repgradelbo_locationscale_bijectors.jl‎ ‎…s/repgradelbo_locationscale_bijectors.jl‎test/algorithms/paramspacesgd/repgradelbo_locationscale_bijectors.jl renamed to test/algorithms/repgradelbo_locationscale_bijectors.jl b/‎…d/repgradelbo_locationscale_bijectors.jl‎ ‎…s/repgradelbo_locationscale_bijectors.jl‎test/algorithms/paramspacesgd/repgradelbo_locationscale_bijectors.jl renamed to test/algorithms/repgradelbo_locationscale_bijectors.jl
diff --git a/‎…gd/repgradelbo_proximal_locationscale.jl‎ ‎…ms/repgradelbo_proximal_locationscale.jl‎test/algorithms/paramspacesgd/repgradelbo_proximal_locationscale.jl renamed to test/algorithms/repgradelbo_proximal_locationscale.jl b/‎…gd/repgradelbo_proximal_locationscale.jl‎ ‎…ms/repgradelbo_proximal_locationscale.jl‎test/algorithms/paramspacesgd/repgradelbo_proximal_locationscale.jl renamed to test/algorithms/repgradelbo_proximal_locationscale.jl
diff --git a/‎…elbo_proximal_locationscale_bijectors.jl‎ ‎…elbo_proximal_locationscale_bijectors.jl‎test/algorithms/paramspacesgd/repgradelbo_proximal_locationscale_bijectors.jl renamed to test/algorithms/repgradelbo_proximal_locationscale_bijectors.jl b/‎…elbo_proximal_locationscale_bijectors.jl‎ ‎…elbo_proximal_locationscale_bijectors.jl‎test/algorithms/paramspacesgd/repgradelbo_proximal_locationscale_bijectors.jl renamed to test/algorithms/repgradelbo_proximal_locationscale_bijectors.jl
diff --git a/‎…lgorithms/paramspacesgd/scoregradelbo.jl‎ ‎test/algorithms/scoregradelbo.jl‎test/algorithms/paramspacesgd/scoregradelbo.jl renamed to test/algorithms/scoregradelbo.jl b/‎…lgorithms/paramspacesgd/scoregradelbo.jl‎ ‎test/algorithms/scoregradelbo.jl‎test/algorithms/paramspacesgd/scoregradelbo.jl renamed to test/algorithms/scoregradelbo.jl
diff --git a/‎…mspacesgd/scoregradelbo_locationscale.jl‎ ‎…lgorithms/scoregradelbo_locationscale.jl‎test/algorithms/paramspacesgd/scoregradelbo_locationscale.jl renamed to test/algorithms/scoregradelbo_locationscale.jl b/‎…mspacesgd/scoregradelbo_locationscale.jl‎ ‎…lgorithms/scoregradelbo_locationscale.jl‎test/algorithms/paramspacesgd/scoregradelbo_locationscale.jl renamed to test/algorithms/scoregradelbo_locationscale.jl
diff --git a/‎…scoregradelbo_locationscale_bijectors.jl‎ ‎…scoregradelbo_locationscale_bijectors.jl‎test/algorithms/paramspacesgd/scoregradelbo_locationscale_bijectors.jl renamed to test/algorithms/scoregradelbo_locationscale_bijectors.jl b/‎…scoregradelbo_locationscale_bijectors.jl‎ ‎…scoregradelbo_locationscale_bijectors.jl‎test/algorithms/paramspacesgd/scoregradelbo_locationscale_bijectors.jl renamed to test/algorithms/scoregradelbo_locationscale_bijectors.jl
diff --git a/‎…lgorithms/paramspacesgd/subsampledobj.jl‎ ‎test/algorithms/subsampledobj.jl‎test/algorithms/paramspacesgd/subsampledobj.jl renamed to test/algorithms/subsampledobj.jl
Lines changed: 1 addition & 0 deletions b/‎…lgorithms/paramspacesgd/subsampledobj.jl‎ ‎test/algorithms/subsampledobj.jl‎test/algorithms/paramspacesgd/subsampledobj.jl renamed to test/algorithms/subsampledobj.jl
Lines changed: 1 addition & 0 deletions
@@ -277,7 +277,6 @@ export optimize
 include("utils.jl")
 include("optimize.jl")
 
-
 ## Parameter Space SGD Implementations
 
 include("algorithms/abstractobjective.jl")
 
@@ -64,6 +64,7 @@ end
         T = 128
         sub = ReshufflingBatchSubsampling(1:n_data, 1)
         alg = KLMinRepGradDescent(AD; subsampling=sub)
+        sub_obj = alg.objective
 
         rng = StableRNG(seed)
         q_avg, _, _ = optimize(rng, alg, T, prob, q0; show_progress=false)