-
-
Notifications
You must be signed in to change notification settings - Fork 63
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
Add threading for variational/optimize #369
Conversation
R/model.R
Outdated
@@ -1024,7 +1029,23 @@ optimize_method <- function(data = NULL, | |||
algorithm = NULL, | |||
init_alpha = NULL, | |||
iter = NULL, | |||
sig_figs = NULL) { | |||
sig_figs = NULL, | |||
threads = NULL) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Not so sure about the name here. but threads_per_chain
does not work for optimize/variational.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yeah that's a good question. Ideally we would use the same argument name but you're right that there are no chains here. I guess threads
is fine. @mitzimorris or @bbbales2 any thoughts on what name we should use here?
I put a few comments, but this looks good to me, thanks! |
Thanks. I guess the only thing to discuss is whether |
I think |
Codecov Report
@@ Coverage Diff @@
## master #369 +/- ##
==========================================
- Coverage 90.58% 88.37% -2.22%
==========================================
Files 12 12
Lines 2571 2614 +43
==========================================
- Hits 2329 2310 -19
- Misses 242 304 +62
Continue to review full report at Codecov.
|
I'm happy with threads. I hope in the future we do a multiple-runs sorta VI (everything should be checked -- https://arxiv.org/abs/2009.00666), but that isn't here now. |
Yeah I think multiple-run VI would be a good thing in the future. I guess we can adjust as necessary when that time comes. |
# Conflicts: # NEWS.md
Seems like we're going with |
Summary
Fixes #193
Copyright and Licensing
Please list the copyright holder for the work you are submitting
(this will be you or your assignee, such as a university or company):
Rok Češnovar, Uni. of Ljubljana
By submitting this pull request, the copyright holder is agreeing to
license the submitted work under the following licenses: