diff --git a/setup.py b/setup.py
index aeed75d..d5fe911 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@
 setup(
     name='swin-transformer-pytorch',
     packages=find_packages(),
-    version='0.3',
+    version='0.4',
     license='MIT',
     description='Swin Transformer - Pytorch',
     long_description=long_description,