Update: include NM_INTEGRATED

neuralmagic · Apr 12, 2022 · fb589b9 · fb589b9
1 parent 8fca412
commit fb589b9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/build_wheel.sh b/build_wheel.sh
@@ -5,6 +5,7 @@ grep --include=\*.py -rnl 'yolov5/' -e "from models" | xargs -i@ sed -i 's/from
 grep --include=\*.py -rnl 'yolov5/' -e "from utils" | xargs -i@ sed -i 's/from utils/from yolov5.utils/g' @
 sed -i '$d' yolov5/requirements.txt
 cat > yolov5/__init__.py << EOF
+from yolov5.utils.general import NM_INTEGRATED
 EOF
 python3 -m build
 rm -r yolov5
diff --git a/utils/general.py b/utils/general.py
@@ -40,6 +40,7 @@
 NUM_THREADS = min(8, max(1, os.cpu_count() - 1))  # number of YOLOv5 multiprocessing threads
 VERBOSE = str(os.getenv('YOLOv5_VERBOSE', True)).lower() == 'true'  # global verbose mode
 FONT = 'Arial.ttf'  # https://ultralytics.com/assets/Arial.ttf
+NM_INTEGRATED=True
 
 torch.set_printoptions(linewidth=320, precision=5, profile='long')
 np.set_printoptions(linewidth=320, formatter={'float_kind': '{:11.5g}'.format})  # format short g, %precision=5