diff --git a/segmentation_models_pytorch/__version__.py b/segmentation_models_pytorch/__version__.py
index cfb0b13d..58114fa8 100644
--- a/segmentation_models_pytorch/__version__.py
+++ b/segmentation_models_pytorch/__version__.py
@@ -1,3 +1,3 @@
-VERSION = (0, 3, 0)
+VERSION = (0, 3, 1)
 
 __version__ = ".".join(map(str, VERSION))