How to modify the source code to allow training on a single machine with 2 CPUs?

dreamquark-ai / tabnet

PyTorch implementation of TabNet paper : https://arxiv.org/pdf/1908.07442.pdf

https://dreamquark-ai.github.io/tabnet/

MIT License

2.61k stars 485 forks source link

How to modify the source code to allow training on a single machine with 2 CPUs? #439

Closed WanBenLe closed 2 years ago

WanBenLe commented 2 years ago

I have 2 XEON 5220R gold CPUs in my Ubuntu system, but during training I found that only one CPU was being used, how can I change this or will it be supported in the future? Thanks.

Optimox commented 2 years ago

You might want to try to change num_workers to 2 and see if things are better.