diff --git a/pytorch_tabular/__init__.py b/pytorch_tabular/__init__.py
index 0ec71f6b..e06ef0a0 100644
--- a/pytorch_tabular/__init__.py
+++ b/pytorch_tabular/__init__.py
@@ -2,7 +2,7 @@
 
 __author__ = """Manu Joseph"""
 __email__ = "manujosephv@gmail.com"
-__version__ = "0.4.0"
+__version__ = "0.5.0-dev0"
 
 from . import models
 from .tabular_datamodule import TabularDatamodule
diff --git a/setup.cfg b/setup.cfg
index 4d73b841..e343ff99 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.4.0
+current_version = 0.5.0-dev0
 commit = True
 tag = False
 parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(\-(?P<release>[a-z]+)(?P<build>\d+))?
diff --git a/setup.py b/setup.py
index b522a628..985ef894 100644
--- a/setup.py
+++ b/setup.py
@@ -58,6 +58,6 @@ def read_requirements(thelibFolder, filename):
     test_suite="tests",
     tests_require=test_requirements,
     url="https://github.com/manujosephv/pytorch_tabular",
-    version="0.4.0",
+    version="0.5.0-dev0",
     zip_safe=False,
 )