更新 'inference.py'

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

View File

@ -89,7 +89,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'