Specify num_threads as a function argument

KarelDO / xmc.dspy

In-Context Learning for eXtreme Multi-Label Classification (XMC) using only a handful of examples.

MIT License

339 stars 19 forks source link

Specify num_threads as a function argument #9

Open abedini-arteriaai opened 3 months ago

abedini-arteriaai commented 3 months ago

In optimizer.py self.num_threads is manually set to 8 so the program errors out when run on a single-thread CPU. Allowing the user to specify num_threads to get passed as a function argument during optimization will fix this issue.