Changed ultralytics version to last DDP stable

Surveily · Jun 7, 2024 · 90a6b5c · 90a6b5c
1 parent aab564b
commit 90a6b5c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/surveily.developer.ai/base/Dockerfile b/src/surveily.developer.ai/base/Dockerfile
@@ -45,11 +45,17 @@ RUN apt-get -yq update \
 
 USER vscode
 
+# WARNING!
+# Change ultralytics version with great care, due
+# to different problems:
+# - DDP
+# - DDP logging callbacks
+# - Runtime parameters
 RUN python3 -m pip install -U pip && \
     python3 -m pip install --user --quiet --no-cache-dir \
     torch==2.0.0 \
     torchvision==0.15.1 \
-    ultralytics==8.2.19 \
+    ultralytics==8.0.202 \
     huggingface_hub==0.18.0
 
 RUN python3 -m pip install --user --quiet --no-cache-dir \