Merge pull request #753 from mlcommons/752_hotfix_for_training

Fixed the issue where a specific library prevents training to start
mlcommons · Dec 1, 2023 · 38fa03b · 38fa03b
2 parents af5af57 + aef322d
commit 38fa03b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -110,7 +110,7 @@ def run(self):
     "pydicom",
     "onnx",
     "torchinfo==1.7.0",
-    "segmentation-models-pytorch==0.3.2",
+    "segmentation-models-pytorch==0.3.3",
     "ACSConv==0.1.1",
     "docker",
     "dicom-anonymizer",