Fix dpn exportable flag

qubvel · qubvel · commit d3cefdda479b · 2025-01-30T09:41:43.000Z
diff --git a/segmentation_models_pytorch/encoders/dpn.py b/segmentation_models_pytorch/encoders/dpn.py
@@ -34,7 +34,7 @@
 
 class DPNEncoder(DPN, EncoderMixin):
     _is_torch_scriptable = False
-    _is_torch_exportable = False
+    _is_torch_exportable = True  # since torch 2.6.0
 
     def __init__(
         self,
