MilesCranmer commited on
Commit
0557713
1 Parent(s): 3a578d2

Reduce number of threads per worker

Browse files
Files changed (2) hide show
  1. julia/sr.jl +1 -1
  2. pysr/sr.py +1 -1
julia/sr.jl CHANGED
@@ -1,5 +1,5 @@
1
  import Optim
2
- using Printf
3
 
4
  const maxdegree = 2
5
  const actualMaxsize = maxsize + maxdegree
 
1
  import Optim
2
+ import Printf: @printf
3
 
4
  const maxdegree = 2
5
  const actualMaxsize = maxsize + maxdegree
pysr/sr.py CHANGED
@@ -213,7 +213,7 @@ const weights = convert(Array{Float32, 1}, """f"{weight_str})"
213
  command = [
214
  f'julia -O{julia_optimization:d}',
215
  f'-p {procs}',
216
- f'--threads auto',
217
  f'/tmp/.runfile_{rand_string}.jl',
218
  ]
219
  if timeout is not None:
 
213
  command = [
214
  f'julia -O{julia_optimization:d}',
215
  f'-p {procs}',
216
+ f'--threads 2', #Threading within each worker
217
  f'/tmp/.runfile_{rand_string}.jl',
218
  ]
219
  if timeout is not None: