Merge pull request #584 from mlrun/1.6.x-dev

1.6.x dev
mlrun · Apr 2, 2024 · dce60a8 · dce60a8
2 parents 8d98b67 + e2cfc03
commit dce60a8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mask-detection/1-training-and-evaluation.ipynb b/mask-detection/1-training-and-evaluation.ipynb
@@ -114,7 +114,8 @@
     "else:\n",
     "    commands = ['pip install torch==1.13.0+cpu torchvision==0.14.0+cpu -f https://download.pytorch.org/whl/torch_stable.html',\n",
     "                'pip install tensorboard==2.5.0',\n",
-    "                'pip install horovod==0.25.0']\n",
+    "                'pip install horovod==0.25.0',\n",
+    "                'pip install onnx~=1.15.0']\n",
     "    builder_env={'HOROVOD_WITH_MPI':'1',\n",
     "                 'HOROVOD_WITH_PYTORCH': '1'}\n",
     "\n",