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.

Sign up for GitHub

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

Jump to bottom

Add a simple PyTorch training example #3327

Merged
merged 6 commits into from
Feb 15, 2024
Merged

Add a simple PyTorch training example #3327

merged 6 commits into from
Feb 15, 2024

Conversation

eb8680
Copy link
Member

@eb8680 eb8680 commented Feb 15, 2024

Resolves #3324

This PR adds a simple example script forked from the existing Lightning example that demonstrates the use of torch.optim optimizers with Pyro's SVI tools.

@eb8680 eb8680 requested a review from fritzo February 15, 2024 18:30
@eb8680 eb8680 added this to the 1.9 release milestone Feb 15, 2024
fritzo
fritzo approved these changes Feb 15, 2024
View reviewed changes
Copy link
Member

@fritzo fritzo left a comment

Choose a reason for hiding this comment

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

Nice, Thanks for adding this!

@fritzo fritzo merged commit 98441c3 into dev Feb 15, 2024
9 checks passed
@eb8680 eb8680 deleted the eb-svi-torch-example branch February 15, 2024 21:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@fritzo fritzo fritzo approved these changes

Assignees
No one assigned
Labels
awaiting review Examples
Projects
None yet
Milestone
1.9 release
Development

Successfully merging this pull request may close these issues.

[FR] Tutorial on PyroModule + ELBO module training with torch optimizers
2 participants
@eb8680 @fritzo