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] マルチスレッド対応 #37

Closed
wants to merge 4 commits into from
Closed

[Feature] マルチスレッド対応 #37

wants to merge 4 commits into from

Conversation

lsz05
Copy link
Collaborator

@lsz05 lsz05 commented Jun 20, 2024

関連する Issue / PR

#36

PR をマージした後の挙動の変化

  • テキストがlist and (deviceが設定されている or multi-GPU環境がある)場合,SentenceTransformer.encode_multi_processを使用
  • その他の場合はSentenceTransformer.encodeのまま

挙動の変更を達成するために行ったこと

  • multiprocessを使用する要件判断
  • multiprocessとmonoprocessの処理を分ける

動作確認

  • テストが通ることを確認した
  • マージ先がdevブランチであることを確認した

@lsz05 lsz05 marked this pull request as ready for review June 24, 2024 08:46
@lsz05
Copy link
Collaborator Author

lsz05 commented Jun 25, 2024

走らせてみたところ,速度が微妙ですので実装変える予定です。

@lsz05 lsz05 closed this Jun 25, 2024
@lsz05 lsz05 changed the title [Fix] マルチスレッド対応 [Feature] マルチスレッド対応 Jun 25, 2024
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.

1 participant