Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Add Support for Multi-GPU Training with PyTorch Lightning #2337

Closed
wants to merge 3 commits into from
Closed

Add Support for Multi-GPU Training with PyTorch Lightning #2337

wants to merge 3 commits into from

Conversation

OrjwanZaafarani
Copy link

This pull request introduces support for multi-GPU training in the Sentence Transformers library using PyTorch Lightning.

The following changes have been made:

  • Updated README.md to include instructions on how to perform multi-GPU training.
  • Added a new module, SentenceTransformerMultiGPU.py, to enable multi-GPU training.
  • Included PyTorch Lightning in the requirements.txt file to ensure compatibility.

This enhancement will allow users to leverage the power of multiple GPUs for faster and more efficient training with Sentence Transformers.

Closes the following issues:

# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant