-
-
Notifications
You must be signed in to change notification settings - Fork 374
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Implement automatic selection of dense/diag metric at warmup #2814
Comments
Is this issue for the sampler and a services function? Could you explain where the sampler functionality will go and how the new services function will differ from our existing dense and diagonal service functions. |
Both and also whatever needs to happen with cmdstan.
I made a third metric option in cmdstan ('auto'). I made a new service function here: https://github.com/stan-dev/stan/blob/88225d08e30636073a310e9cc4eefc97e5212dbf/src/stan/services/sample/hmc_nuts_auto_e_adapt.hpp To give some idea of what changed, the changed file list is:
|
Summary:
Implement the automatic selection between metric=dense_e and metric=diag_e as described in: https://arxiv.org/abs/1905.11916
Description
It's not always obvious when the diagonal or dense metrics are the best. Or even if it's obvious that a dense metric will be good eventually, early in warmup the proposed metrics might not be so great.
This change will, at the end of each warmup stage where the metric is recomputed, decide whether to use a dense or diagonal metric for the next warmup stage (or for the sampling) automatically.
Current Version:
v2.20.0
The text was updated successfully, but these errors were encountered: