feat: change argument gpu to store_true

mumu0419 · Dec 29, 2020 · b9cca00 · b9cca00
1 parent 0df7238
commit b9cca00
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test.py b/test.py
@@ -24,7 +24,7 @@
     parser = argparse.ArgumentParser()
     parser.add_argument('-net', type=str, required=True, help='net type')
     parser.add_argument('-weights', type=str, required=True, help='the weights file you want to test')
-    parser.add_argument('-gpu', type=bool, default=True, help='use gpu or not')
+    parser.add_argument('-gpu', action='store_true', default=False, help='use gpu or not')
     parser.add_argument('-b', type=int, default=16, help='batch size for dataloader')
     args = parser.parse_args()