-
Notifications
You must be signed in to change notification settings - Fork 917
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
fix documentation to match pytorch-lightning #1244
fix documentation to match pytorch-lightning #1244
Conversation
Codecov ReportBase: 94.02% // Head: 93.97% // Decreases project coverage by
Additional details and impacted files@@ Coverage Diff @@
## master #1244 +/- ##
==========================================
- Coverage 94.02% 93.97% -0.05%
==========================================
Files 73 73
Lines 8215 8182 -33
==========================================
- Hits 7724 7689 -35
- Misses 491 493 +2
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. ☔ View full report at Codecov. |
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.
Thanks again for another PR :)
We would have to adapt the parts of the code where we generate params dicts for PyTorch Lightning Trainer with "gpus". Otherwise the deprecation warning will not disappear.
I added a suggestion.
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.
LGTM, however we could consider moving these comments to actual docstring, and remove the support for torch_device_str
, WDYT?
Agreed 👍 |
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.
LGTM, thanks @gdevos010 !
Just have a small question about the pytorch lightning version (>=1.7.0) check in torch_forecasting_model.py
: is this needed?
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.
Nice :) Only thing is that we should set the 3 parameters by default "accelerator", "gpus" (or "devices"), "auto_select_gpus", as we decided to always use CPU by default.
After that we can merge :)
@dennisbader we should be good to merge now |
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, looks perferct now 👍 Thanks a lot for all the effort @gdevos010!
the
gpus
arg in PyTorch Lightning has been deprecated in favor ofdevices
fixed broken PyTorch Lightning link
https://pytorch-lightning.readthedocs.io/en/stable/common/trainer.html#gpus