diff --git a/.bumpversion.cfg b/.bumpversion.cfg
index 93a2c82..b3b3da2 100644
--- a/.bumpversion.cfg
+++ b/.bumpversion.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.5.3
+current_version = 0.5.4
 commit = True
 tag = True
 parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)
diff --git a/docs/source/conf.py b/docs/source/conf.py
index 3b1f5de..656b123 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -23,7 +23,7 @@ author = "Jensun Ravichandran"
 
 # The full version, including alpha/beta/rc tags
 #
-release = "0.5.3"
+release = "0.5.4"
 
 # -- General configuration ---------------------------------------------------
 
diff --git a/prototorch/models/__init__.py b/prototorch/models/__init__.py
index ff28b0a..42e5ca8 100644
--- a/prototorch/models/__init__.py
+++ b/prototorch/models/__init__.py
@@ -36,4 +36,4 @@ from .unsupervised import (
 )
 from .vis import *
 
-__version__ = "0.5.3"
+__version__ = "0.5.4"
diff --git a/setup.py b/setup.py
index c2a4df8..bb4b478 100644
--- a/setup.py
+++ b/setup.py
@@ -55,7 +55,7 @@ ALL = CLI + DEV + DOCS + EXAMPLES + TESTS
 
 setup(
     name=safe_name("prototorch_" + PLUGIN_NAME),
-    version="0.5.3",
+    version="0.5.4",
     description="Pre-packaged prototype-based "
     "machine learning models using ProtoTorch and PyTorch-Lightning.",
     long_description=long_description,