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

feature: add support for PyTorch Lightning #774

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

clumsy
Copy link

@clumsy clumsy commented Mar 19, 2024

Issue #, if available: N/A

Description of changes: Adds PyTorch Lightning support, LightningModule and LightningDataModule for Node GNN model and also a Jupyter nodebook demonstrating how it works.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@thvasilo
Copy link
Contributor

thvasilo commented Jun 4, 2024

Hello @clumsy took a closer look at the PR and had a couple of requests. We'd like to maintain a similar level of requirements between GSProcessing and this library.

  • We should try to have some unit tests if possible.
  • Let's put some more work pyproject.toml. Right now in the dev group I see a lot of dependencies that I'm not sure the project needs. For DGL in particular there's only support for py3.10 on Linux?
  • Let's add some documentation. The example notebook is helpful but we also want to have a guide that helps users from installation to training, with AWS as the target cloud infra. At least one example on how to run this on SageMaker should be included.
  • This also means that we likely have to include a compatible Dockerfile. If the existing SageMaker and local Dockerfiles can be slightly modified to add PTL support, we can have that in documentation.

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.

3 participants