google-research / smore

Apache License 2.0
162 stars 28 forks source link

Confusion of the --cpu_num flag #14

Open Juanhui28 opened 1 year ago

Juanhui28 commented 1 year ago

Hi there,

We are running the train_concat_wikikgv2.sh in the training/complex_scripts folder, when we use the default setting (--cpu_num=8), it actually requires around 60 cpu cores which makes the code to be killed easily. And when I tried to reduce the number (even to zero), it's still get killed.

So I'm a little bit confused how can we set the number cpu cores? Any help is appreciated!

Thank you!