-
Notifications
You must be signed in to change notification settings - Fork 3.4k
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 LightningLite documentation #10043
Conversation
Build Error! No Linked Issue found. Please link an issue or mention it in the body using #<issue_id> |
Also, one more thing we could add to docs is How a User could power up their PyTorch Loops with DeepSpeed, Fairscale and XLA without dealing with their APIs. |
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 awesome 🚀 . just some small comments
def training_step(self, batch, batch_idx): | ||
x = self.forward(batch) | ||
self.log("train_loss", x) | ||
return x | ||
|
||
def validation_step(self, batch, batch_idx): | ||
x = self.forward(batch) | ||
self.log("val_loss", x) | ||
return x |
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.
x
isn't loss right?
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.
Yes, it is.
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.
so trainer.predict
will return losses? since it just return what forward
returns by default.
Merging this now into POC. We can keep fine tuning it. |
Awesome!!! |
What does this PR do?
This PR adds documentation for
LightningLite
Parts of #9987
Fixes #8620
Does your PR introduce any breaking changes? If yes, please list them.
Before submitting
PR review
Anyone in the community is welcome to review the PR.
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 🙃