From 9f7d9570ae43eedeb11f89da92bf4b28ae38ac03 Mon Sep 17 00:00:00 2001 From: jeehyunHwang Date: Fri, 5 Nov 2021 18:52:43 +0900 Subject: [PATCH] minor fix on main --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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