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

[FEAT] Add support for lightning>=2.0.0, and torch>=2.0.0 #498

Merged
merged 9 commits into from
Apr 4, 2023

Conversation

AzulGarza
Copy link
Member

This PR upgrades lightning and torch to the most recent versions. The main change is validation_epoch_end -> on_validation_epoch_end (now, the validation outputs need to be stored in the validation_step to be used in on_validation_epoch_end). See this PR.

Also, this PR deprecates python3.7 (the minimum version compatible with lightning>=2.0.0 is py3.8).

@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@AzulGarza AzulGarza requested a review from cchallu April 3, 2023 19:10
@cchallu cchallu merged commit e9270e8 into main Apr 4, 2023
8 checks passed
@cchallu cchallu deleted the feat/lightning-torch-2 branch April 4, 2023 17:30
@candalfigomoro
Copy link
Contributor

@FedericoGarza

This could make it difficult to use the library on Google Vertex AI Workbench (the Jupyter notebook of Google Cloud Platform), because the default environment uses Python 3.7 and upgrading the Python version is a bit tricky (although I hope it will be upgraded before Python 3.7 is end-of-life in June).

@AzulGarza
Copy link
Member Author

hey @candalfigomoro! I've tried to support Python 3.7 here but unfortunately, torch no longer supports it. :( The workaround would be to fix the version to the latest stable release (neuralforecast==1.4.0 and torch<2.0.0).

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.

None yet

3 participants