chenyh
/
MNIST_Example
mirror of https://openi.pcl.ac.cn/OpenIOSSG/MNIST_Example.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

更新 '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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference.py
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)') help='device where the code will be implemented (default: Ascend)')
if __name__ == "__main__": if __name__ == "__main__":
args = parser.parse_args() args, unknown = parser.parse_known_args()
###Initialize the data and result directories in the inference image### ###Initialize the data and result directories in the inference image###
data_dir = '/cache/data' data_dir = '/cache/data'