minor fix on main

jeehyunHwang · Nov 5, 2021 · 9f7d957 · 9f7d957
1 parent 1cfb5d3
commit 9f7d957
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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