ATC工具yolov5 onnx模型转om失败
收藏回复举报
ATC工具yolov5 onnx模型转om失败
t('forum.solved') 已解决
发表于2024-10-18 11:58:04
0 查看

(base) root@davinci-mini:/home/HwHiAiUser/samples/notebooks/DeepSort# atc --model=best.onnx --framework=5 --output=yolov5s --input_format=NCHW --log=error --soc_version=Ascend310B1 --input_shape="images:1,3,640,640"



ATC start working now, please wait for a moment.



Exception in thread Thread-1:



Traceback (most recent call last):



  File "/usr/local/miniconda3/lib/python3.9/threading.py", line 954, in _bootstrap_inner



    self.run()



  File "/usr/local/Ascend/ascend-toolkit/latest/python/site-packages/tbe/common/repository_manager/utils/multiprocess_util.py", line 91, in run



    key, func, args, kwargs = self.task_q.get(timeout=TIMEOUT)



  File "<string>", line 2, in get



  File "/usr/local/miniconda3/lib/python3.9/multiprocessing/managers.py", line 809, in _callmethod



    kind, result = conn.recv()



  File "/usr/local/miniconda3/lib/python3.9/multiprocessing/connection.py", line 255, in recv



    buf = self._recv_bytes()



  File "/usr/local/miniconda3/lib/python3.9/multiprocessing/connection.py", line 419, in _recv_bytes



    buf = self._recv(4)



  File "/usr/local/miniconda3/lib/python3.9/multiprocessing/connection.py", line 384, in _recv



    chunk = read(handle, remaining)



ConnectionResetError: [Errno 104] Connection reset by peer



Traceback (most recent call last):



  File "/usr/local/miniconda3/lib/python3.9/multiprocessing/util.py", line 300, in _run_finalizers



    finalizer()



  File "/usr/local/miniconda3/lib/python3.9/multiprocessing/util.py", line 224, in __call__



    res = self._callback(*self._args, **self._kwargs)



  File "/usr/local/Ascend/ascend-toolkit/latest/python/site-packages/tbe/common/repository_manager/route.py", line 51, in wrapper



    return func(cls, *args, **kwargs)



  File "/usr/local/Ascend/ascend-toolkit/latest/python/site-packages/tbe/common/repository_manager/route.py", line 207, in finalize



    cls.bank_process_pool.close()



AttributeError: 'NoneType' object has no attribute 'close'



...



ATC run failed, Please check the detail log, Try 'atc --help' for more information

我要发帖子