limit the number of CPU threads used by pytorch

compbiomed-unito / survhive

Convenient, opinionated wrapper around some (deep) survival models

MIT License

2 stars 0 forks source link

Open IvanUniTO opened 10 months ago

IvanUniTO commented 10 months ago

all the pytorch-based methods now use all of them.

IvanUniTO commented 10 months ago

the use of torch. set_num_threads seems to be working on auton.