Hello, In NHITs, how can I specify the number of cpus to use for training?

Could you be a bit more specific - do you use multiple physical cpus? Or do you mean specify the number of threads the process uses with a single cpu?

I want to specify the number of threads for the process. IF that can be affected by some other parameter including number of cpus please let me know on those parameters as well.

You could do e.g.:

import torch

torch.set_num_threads(10)