diff --git a/setup.py b/setup.py
index 46323e207..464f7a603 100644
--- 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",