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

PyTorch 2.0 #375

Closed
xianyuanliu opened this issue May 3, 2023 · 1 comment
Closed

PyTorch 2.0 #375

xianyuanliu opened this issue May 3, 2023 · 1 comment

Comments

@xianyuanliu
Copy link
Member

Issues may arise in this upgrade. Edited by Xianyuan. (continuous updating)
Upgrading to PyTorch v2.x:

  1. When using PyTorch v2.x, the PyTorch-Lightning setting must also be upgraded. Our current setting is lower than v1.6.5. This version is not compatible with PyTorch v2.0.0, but PyTorch-Lightning v2.0.1 is.

Upgrading to PyTorch-Lightning v2.x:

  1. PyTorch-Lightning v2.0.0 does not support training_epoch_end; the new version uses the on_train_epoch_end hook instead. Detailed examples can be found in Remove memory-retaining epoch-end hooks Lightning-AI/pytorch-lightning#16520. This issue would affect some of the pipeline trainers.
  2. Guide to upgrade to v2.0.
@xianyuanliu
Copy link
Member Author

Moved to #376.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
PyKale v0.2.0
Status: Done
Milestone
No milestone
Development

No branches or pull requests
1 participant
@xianyuanliu