更新 'inference_for_multidataset.py'

兼容输入参数与解析参数不完全一致的情况
This commit is contained in:
ychao_1983 2022-08-30 14:44:25 +08:00
parent d913e45367
commit 789a680fd9
1 changed files with 1 additions and 1 deletions

View File

@ -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'