parent
d913e45367
commit
789a680fd9
|
@ -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'
|
||||
|
|
Loading…
Reference in New Issue