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