how to set threads when use laaonet?

[zhao qiang]

I noticed that the lassonet will use half of threads to run. I wonder to know how to use more threads when running lassonet?

[Louis Abraham]

Do you use the CPU or GPU?

[zhao qiang]

CPU

[Louis Abraham]

What method did you use? a -CV one?

[zhao qiang]

LassoNetClassifierCV

[ilemhadri]

As usual, having a fully reproducible code snippet would probably be helpful.

[Louis Abraham]

Well, there are multiple answers:

• you can control the number of thread used by pytorch: https://jdhao.github.io/2020/07/06/pytorch_set_num_threads/
• you could parallelize the CV loop here:

  lassonet/lassonet/interfaces.py

  Line 663 in e12ca78

   # TODO: parallelize