From 4b4fdb578a3a7ff08828922f4cb1f7f2c32d8ebc Mon Sep 17 00:00:00 2001 From: chengchunlei <0510568@163.com> Date: Sat, 24 Oct 2020 17:47:40 +0800 Subject: [PATCH] update dnmetis/main.py. --- dnmetis/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dnmetis/main.py b/dnmetis/main.py index 2c5880b..6e24e2b 100644 --- a/dnmetis/main.py +++ b/dnmetis/main.py @@ -34,7 +34,7 @@ def get_args(): # below will override DNMetis rules compliant settings - don't use for official submission parser.add_argument("--count", default=1000, type=int, help="dataset items to infer") - parser.add_argument("--precision", default="fp32", choices=["fp32", "fp16", "int8", "uint8"], + parser.add_argument("--precision", default="fp16", choices=["fp32", "fp16", "int8", "uint8"], help="precision mode, one of " + str(["fp32", "fp16", "int8", "uint8"])) parser.add_argument("--feed", default=[], help="feed") parser.add_argument("--image_list", default=[], help="image_list")