瀏覽代碼

修改参数错误

zhy 10 小時之前
父節點
當前提交
da7058ad52
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      tests/verify_tool_accuracy_test.py

+ 1 - 1
tests/verify_tool_accuracy_test.py

@@ -100,7 +100,7 @@ if __name__ == '__main__':
 
             # verify_result = verify_tool.label_verification(onnx_file)
             # 调用验证工具进行标签验证
-            verify_result = verify_tool_mix.label_verification(onnx_file, framework=args.framework, mode=args.mode, model_type=args.model_value)
+            verify_result = verify_tool_mix.label_verification(onnx_file, framework=args.framework, mode=args.mode, model_type=args.model_type)
             total += 1
             if str(verify_result) == args.except_result:
                 correct += 1