jerryji1993 / DNABERT

DNABERT: pre-trained Bidirectional Encoder Representations from Transformers model for DNA-language in genome

Home Page:https://doi.org/10.1093/bioinformatics/btab083

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

how to limit cpu usage

Jasonxu0109 opened this issue · comments

hi,
how to limit cpu and memory usage. Because i have set n_process as 1,but also produce high cpu usage. How to fix it? Thanks

You can try to use torch.set_num_threads() to set the number of CPU cores to use