diff --git a/setup.py b/setup.py
index 03e2a71..c006fe3 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='UniTrain',
-    version='0.0.1',
+    version='0.2',
     author='Ishan Upadhyay',
     author_email='ishan.upadhyay.iitb@gmail.com',
     description='A generalized training framework for Deep Learning Tasks',