diff --git a/main.py b/main.py
index a192d3b..35cb4e1 100644
--- a/main.py
+++ b/main.py
@@ -31,7 +31,7 @@ def get_parser():
     parser.add_argument("--gpu",
                 type=str,
                 help="gpu num",
-                default="1")
+                default="0")
 
     return parser