Pārlūkot izejas kodu

Default `OMP_NUM_THREADS=8` (#6770)

modifyDataloader
Glenn Jocher GitHub pirms 2 gadiem
vecāks
revīzija
c161557563
Šim parakstam datu bāzē netika atrasta zināma atslēga GPG atslēgas ID: 4AEE18F83AFDEB23
1 mainītis faili ar 1 papildinājumiem un 0 dzēšanām
  1. +1
    -0
      utils/general.py

+ 1
- 0
utils/general.py Parādīt failu

@@ -45,6 +45,7 @@ np.set_printoptions(linewidth=320, formatter={'float_kind': '{:11.5g}'.format})
pd.options.display.max_columns = 10
cv2.setNumThreads(0) # prevent OpenCV from multithreading (incompatible with PyTorch DataLoader)
os.environ['NUMEXPR_MAX_THREADS'] = str(NUM_THREADS) # NumExpr max threads
os.environ['OMP_NUM_THREADS'] = str(NUM_THREADS) # OpenMP max threads (PyTorch and SciPy)


def is_kaggle():

Notiek ielāde…
Atcelt
Saglabāt