How to control number of threads?

[SmilingYogi]

I am using grape.embedders.Node2VecGloVeEnsmallen in an environment where there is a strict limit on number of threads being spawned. I could not find any argument that sets the number of threads.

Is there a recommended way to work around this?

[LucaCappelletti94]

Try setting the RAYON_NUM_THREADS system variable.

[SmilingYogi]

Thanks for the reply! Is there any way to do it within the grape functions only?

[LucaCappelletti94]

Not for those using Rayon for parallelism such as Node2VecGloVeEnsmallen. The ones that were straightforwardly possible to set it via a parameter are already available as kwargs.

[SmilingYogi]

Okay. Can you point me to any Node2Vec or similar embedding methods where I can set it via a parameter?

Thanks!