Skip to content

Conversation

@jleinonen
Copy link
Collaborator

Earth2Studio Pull Request

Description

Checklist

  • I am familiar with the Contributing Guidelines.
  • New or existing tests cover these changes.
  • The documentation is up to date with these changes.
  • The CHANGELOG.md is up to date with these changes.
  • An issue is linked to this pull request.

Dependencies

@jleinonen jleinonen requested a review from NickGeneva May 20, 2025 17:11
@jleinonen jleinonen self-assigned this May 20, 2025
@swbg
Copy link
Collaborator

swbg commented May 21, 2025

I am wondering whether it would make sense to support the distributed recipe on a single GPU as well. So users can test the code with a single GPU and then move to more GPUs later.

@jleinonen
Copy link
Collaborator Author

/blossom-ci

@jleinonen
Copy link
Collaborator Author

/blossom-ci

@dallasfoster dallasfoster added the 5 - DO NOT MERGE Hold off on merging; see PR for details label Jun 3, 2025
@dallasfoster dallasfoster self-requested a review June 3, 2025 16:12
@dallasfoster
Copy link
Collaborator

Will review design of distributed utility closely and likely suggest modifications.

@jleinonen
Copy link
Collaborator Author

NOTE: It was reported to me that this had problems with PyTorch 2.7 so we should test with that and fix the problem if necessary before merging.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

5 - DO NOT MERGE Hold off on merging; see PR for details

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants