diff --git a/setup.py b/setup.py
index f4399104..feac8914 100644
--- a/setup.py
+++ b/setup.py
@@ -73,7 +73,7 @@ def get_device_cc():
 
 setup(
     name='bmtrain',
-    version='0.1.5',
+    version='0.1.6',
     author="Guoyang Zeng",
     author_email="qbjooo@qq.com",
     description="A toolkit for training big models",