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 reinforcement learning example for Fabric #16506
Conversation
for more information, see https://pre-commit.ci
…tning into docs/fabric-ppo-rl
…tning into docs/fabric-ppo-rl
for more information, see https://pre-commit.ci
…tning into docs/fabric-ppo-rl
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.
Amazing. Thank you for contributing this example @belerico
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.
Looks great, small tweak to the intro
Co-authored-by: Luca Antiga <luca@lightning.ai>
Great job @belerico ! |
Thank you all! |
Sweet, nice first contribution! 🎉 |
What does this PR do?
This PR introduces a first example of Reinforcement Learning training accelerated by Fabric. In particular, the Proximal Policy Optimization (PPO) algorithm is implemented. As for the other examples, two scripts are given: one that runs the optimization with raw PyTorch, while the other one employing Fabric.
Does your PR introduce any breaking changes? If yes, please list them.
Before submitting
PR review
Anyone in the community is free to review the PR once the tests have passed.
Before you start reviewing make sure you have read Review guidelines. In short, see the following bullet-list:
Did you have fun?
Make sure you had fun coding 🙃