diff --git a/src/stan/mcmc/hmc/nuts/adapt_dense_e_nuts.hpp b/src/stan/mcmc/hmc/nuts/adapt_dense_e_nuts.hpp index 6f0e43c6e6b..377cd96eab9 100644 --- a/src/stan/mcmc/hmc/nuts/adapt_dense_e_nuts.hpp +++ b/src/stan/mcmc/hmc/nuts/adapt_dense_e_nuts.hpp @@ -35,10 +35,10 @@ class adapt_dense_e_nuts : public dense_e_nuts, = this->covar_adaptation_.learn_covariance(inv_metric, this->z_.q); if (update) { - this->init_stepsize(logger); - this->z_.set_inv_metric(inv_metric); + this->init_stepsize(logger); + this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_)); this->stepsize_adaptation_.restart(); } diff --git a/src/stan/mcmc/hmc/nuts/adapt_diag_e_nuts.hpp b/src/stan/mcmc/hmc/nuts/adapt_diag_e_nuts.hpp index 984602a91a4..33cd61e13f1 100644 --- a/src/stan/mcmc/hmc/nuts/adapt_diag_e_nuts.hpp +++ b/src/stan/mcmc/hmc/nuts/adapt_diag_e_nuts.hpp @@ -35,10 +35,10 @@ class adapt_diag_e_nuts : public diag_e_nuts, = this->var_adaptation_.learn_variance(inv_metric, this->z_.q); if (update) { - this->init_stepsize(logger); - this->z_.set_inv_metric(inv_metric); + this->init_stepsize(logger); + this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_)); this->stepsize_adaptation_.restart(); } diff --git a/src/stan/mcmc/hmc/nuts_classic/adapt_dense_e_nuts_classic.hpp b/src/stan/mcmc/hmc/nuts_classic/adapt_dense_e_nuts_classic.hpp index 0245d78d954..d930bb5bec5 100644 --- a/src/stan/mcmc/hmc/nuts_classic/adapt_dense_e_nuts_classic.hpp +++ b/src/stan/mcmc/hmc/nuts_classic/adapt_dense_e_nuts_classic.hpp @@ -35,10 +35,10 @@ class adapt_dense_e_nuts_classic : public dense_e_nuts_classic, = this->covar_adaptation_.learn_covariance(inv_metric, this->z_.q); if (update) { - this->init_stepsize(logger); - this->z_.set_inv_metric(inv_metric); + this->init_stepsize(logger); + this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_)); this->stepsize_adaptation_.restart(); } diff --git a/src/stan/mcmc/hmc/nuts_classic/adapt_diag_e_nuts_classic.hpp b/src/stan/mcmc/hmc/nuts_classic/adapt_diag_e_nuts_classic.hpp index 56450a86ab9..14ab69a04cc 100644 --- a/src/stan/mcmc/hmc/nuts_classic/adapt_diag_e_nuts_classic.hpp +++ b/src/stan/mcmc/hmc/nuts_classic/adapt_diag_e_nuts_classic.hpp @@ -36,10 +36,10 @@ class adapt_diag_e_nuts_classic : public diag_e_nuts_classic, = this->var_adaptation_.learn_variance(inv_metric, this->z_.q); if (update) { - this->init_stepsize(logger); - this->z_.set_inv_metric(inv_metric); + this->init_stepsize(logger); + this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_)); this->stepsize_adaptation_.restart(); } diff --git a/src/stan/mcmc/hmc/static/adapt_dense_e_static_hmc.hpp b/src/stan/mcmc/hmc/static/adapt_dense_e_static_hmc.hpp index c96cd34ca2a..119ed9a5f97 100644 --- a/src/stan/mcmc/hmc/static/adapt_dense_e_static_hmc.hpp +++ b/src/stan/mcmc/hmc/static/adapt_dense_e_static_hmc.hpp @@ -38,10 +38,10 @@ class adapt_dense_e_static_hmc : public dense_e_static_hmc, = this->covar_adaptation_.learn_covariance(inv_metric, this->z_.q); if (update) { - this->init_stepsize(logger); - this->update_L_(); this->z_.set_inv_metric(inv_metric); + this->init_stepsize(logger); + this->update_L_(); this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_)); this->stepsize_adaptation_.restart(); } diff --git a/src/stan/mcmc/hmc/static/adapt_diag_e_static_hmc.hpp b/src/stan/mcmc/hmc/static/adapt_diag_e_static_hmc.hpp index a775a0417ed..fe8f39c25c0 100644 --- a/src/stan/mcmc/hmc/static/adapt_diag_e_static_hmc.hpp +++ b/src/stan/mcmc/hmc/static/adapt_diag_e_static_hmc.hpp @@ -38,10 +38,10 @@ class adapt_diag_e_static_hmc : public diag_e_static_hmc, = this->var_adaptation_.learn_variance(inv_metric, this->z_.q); if (update) { - this->init_stepsize(logger); - this->update_L_(); this->z_.set_inv_metric(inv_metric); + this->init_stepsize(logger); + this->update_L_(); this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_)); this->stepsize_adaptation_.restart(); } diff --git a/src/stan/mcmc/hmc/static_uniform/adapt_dense_e_static_uniform.hpp b/src/stan/mcmc/hmc/static_uniform/adapt_dense_e_static_uniform.hpp index 898036f7925..0af604faf1e 100644 --- a/src/stan/mcmc/hmc/static_uniform/adapt_dense_e_static_uniform.hpp +++ b/src/stan/mcmc/hmc/static_uniform/adapt_dense_e_static_uniform.hpp @@ -37,10 +37,10 @@ class adapt_dense_e_static_uniform = this->covar_adaptation_.learn_covariance(inv_metric, this->z_.q); if (update) { - this->init_stepsize(logger); - this->z_.set_inv_metric(inv_metric); + this->init_stepsize(logger); + this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_)); this->stepsize_adaptation_.restart(); } diff --git a/src/stan/mcmc/hmc/static_uniform/adapt_diag_e_static_uniform.hpp b/src/stan/mcmc/hmc/static_uniform/adapt_diag_e_static_uniform.hpp index 6f5a7356288..b0fd0f5ab84 100644 --- a/src/stan/mcmc/hmc/static_uniform/adapt_diag_e_static_uniform.hpp +++ b/src/stan/mcmc/hmc/static_uniform/adapt_diag_e_static_uniform.hpp @@ -36,10 +36,10 @@ class adapt_diag_e_static_uniform bool update = this->var_adaptation_.learn_variance(inv_metric, this->z_.q); if (update) { - this->init_stepsize(logger); - this->z_.set_inv_metric(inv_metric); + this->init_stepsize(logger); + this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_)); this->stepsize_adaptation_.restart(); } diff --git a/src/stan/mcmc/hmc/xhmc/adapt_dense_e_xhmc.hpp b/src/stan/mcmc/hmc/xhmc/adapt_dense_e_xhmc.hpp index d5b17c432f2..f09186507ec 100644 --- a/src/stan/mcmc/hmc/xhmc/adapt_dense_e_xhmc.hpp +++ b/src/stan/mcmc/hmc/xhmc/adapt_dense_e_xhmc.hpp @@ -35,10 +35,10 @@ class adapt_dense_e_xhmc : public dense_e_xhmc, = this->covar_adaptation_.learn_covariance(inv_metric, this->z_.q); if (update) { - this->init_stepsize(logger); - this->z_.set_inv_metric(inv_metric); + this->init_stepsize(logger); + this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_)); this->stepsize_adaptation_.restart(); } diff --git a/src/stan/mcmc/hmc/xhmc/adapt_diag_e_xhmc.hpp b/src/stan/mcmc/hmc/xhmc/adapt_diag_e_xhmc.hpp index 158c7bb2eda..b28fb886eb6 100644 --- a/src/stan/mcmc/hmc/xhmc/adapt_diag_e_xhmc.hpp +++ b/src/stan/mcmc/hmc/xhmc/adapt_diag_e_xhmc.hpp @@ -35,10 +35,10 @@ class adapt_diag_e_xhmc : public diag_e_xhmc, = this->var_adaptation_.learn_variance(inv_metric, this->z_.q); if (update) { - this->init_stepsize(logger); - this->z_.set_inv_metric(inv_metric); + this->init_stepsize(logger); + this->stepsize_adaptation_.set_mu(log(10 * this->nom_epsilon_)); this->stepsize_adaptation_.restart(); }