checkout branch

uug-ai · Nov 27, 2024 · 29655fe · 29655fe
1 parent 6f002bd
commit 29655fe
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -15,7 +15,9 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
 WORKDIR /usr/src/ultralytics
 
 # Clone the repository
-RUN git clone https://github.com/ultralytics/ultralytics -b main /usr/src/ultralytics
+RUN git clone https://github.com/ultralytics/ultralytics /usr/src/ultralytics
+# Checkout tag v8.1.0
+RUN cd /usr/src/ultralytics && git checkout v8.1.0
 
 # Add yolov8n.pt model
 ADD https://github.com/ultralytics/assets/releases/download/v8.1.0/yolov8n.pt /usr/src/ultralytics/

diff --git a/object_classification_yolov8.py b/object_classification_yolov8.py
@@ -25,7 +25,7 @@
 
 # Following error is thrown: [W NNPACK.cpp:64] Could not initialize NNPACK! Reason: Unsupported hardware.
 # https://stackoverflow.com/questions/69711410/could-not-initialize-nnpack
-torch.backends.nnpack.enabled = False
+# torch.backends.nnpack.enabled = False
 
 # Initialize the VariableClass object, which contains all the necessary environment variables.
 var = VariableClass()