Glenn Jocher
GitHub
преди 2 години
променени са 1 файла, в които са добавени 1 реда и са изтрити 0 реда
+ 1
- 0
utils/general.py
Целия файл
| @@ -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(): | |||
Loading…