diff --git a/luxonis_train/__init__.py b/luxonis_train/__init__.py
index 95051659..ac6e38a1 100644
--- a/luxonis_train/__init__.py
+++ b/luxonis_train/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.1.0"
+__version__ = "0.2.0"
 
 import warnings