Auto3d doesn't set OMP_NUM_THREADS 

**Describe the bug**
Running Auto3d I get warning of 
```
OMP_NUM_THREADS set to 1 to avoid system overload
```

we should smartly export this variable before the torchrun 

I tired exporting it myself but this doesn't work with multiple nodes. I think this parameter needs to be set in the python code itself 
