schihei/ascend-tools
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update dnmetis/main.py.

Browse Source
This commit is contained in:
chengchunlei
2020-10-24 17:47:40 +08:00
committed by Gitee
parent 5e697c736b
commit 4b4fdb578a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dnmetis/main.py
+1 -1
View File
@@ -34,7 +34,7 @@ def get_args():
# below will override DNMetis rules compliant settings - don't use for official submission # 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("--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"])) help="precision mode, one of " + str(["fp32", "fp16", "int8", "uint8"]))
parser.add_argument("--feed", default=[], help="feed") parser.add_argument("--feed", default=[], help="feed")
parser.add_argument("--image_list", default=[], help="image_list") parser.add_argument("--image_list", default=[], help="image_list")