From 789a680fd970d3e8afc13afe3a05193e4ef53ce6 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Tue, 30 Aug 2022 14:44:25 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'inference=5Ffor=5Fmultida?=
 =?UTF-8?q?taset.py'?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

兼容输入参数与解析参数不完全一致的情况
---
 inference_for_multidataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference_for_multidataset.py b/inference_for_multidataset.py
index cf7ae1e..c7f17fa 100644
--- a/inference_for_multidataset.py
+++ b/inference_for_multidataset.py
@@ -108,7 +108,7 @@ parser.add_argument('--device_target', type=str, default="Ascend", choices=['Asc
                     help='device where the code will be implemented (default: Ascend)')                
 
 if __name__ == "__main__":            
-    args = parser.parse_args()
+    args, unknown = parser.parse_known_args()
 
     ###Initialize the data and result directories in the inference image###
     data_dir = '/cache/data'