Skip to content
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

gpytorch.settings.variational_cholesky_jitter can be set dynamically. #2255

Merged
merged 4 commits into from
Mar 6, 2023

Conversation

gpleiss
Copy link
Member

@gpleiss gpleiss commented Jan 17, 2023

Previously, this context manager was only used when VariationalStrategy modules were initialized. With this change,
gpytorch.settings.variational_cholesky_jitter will dynamically change the jitter value (for variational models already in use), unless the user specifies a jitter_val in the VariationalStrategy constructor.

In addition, this PR adds type hintsd to a majority of the VariationalStrategy modules.

[Fixes #2244]

@gpleiss gpleiss requested a review from Balandat January 30, 2023 21:40
Copy link
Collaborator

@Balandat Balandat left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Very nice.

You'll probably have to rebase and rerun the pre-commit hooks with my recent ufmt changes.

Previously, this context manager was only used when VariationalStrategy
modules were initialized. With this change,
gpytorch.settings.variational_cholesky_jitter will dynamically change
the jitter value (for variational models already in use), unless the
user specifies a `jitter_val` in the VariationalStrategy constructor.

In addition, this PR adds type hintsd to a majority of the
VariationalStrategy modules.

[Fixes #2244]
@gpleiss gpleiss force-pushed the dynamic_variational_jitter branch from 497830b to 854df95 Compare March 6, 2023 21:35
@gpleiss gpleiss enabled auto-merge (squash) March 6, 2023 21:35
@gpleiss gpleiss force-pushed the dynamic_variational_jitter branch from 854df95 to be2ce67 Compare March 6, 2023 21:59
@gpleiss gpleiss force-pushed the dynamic_variational_jitter branch from be2ce67 to a10155c Compare March 6, 2023 22:49
@gpleiss gpleiss merged commit 0b50173 into master Mar 6, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[Bug] settings.variational_cholesky_jitter does not work in run time
2 participants