Hi,
I am trying to compile my ONNX model using the Hailo Model Zoo. This ONNX model was retrained with custom data based on yolov5m_wo_spp within the Hailo retrain docker environment. However, I encountered the following error during compilation:
hailomz compile --ckpt /home/ubuntu/best.onnx --calib-path /home/ubuntu/datasets/calib/ --yaml /home/ubuntu/hailo_model_zoo/hailo_model_zoo/cfg/networks/yolov5m_wo_spp.yaml --classes 7
I followed this tutorial to retrain using a custom dataset. Could there be an error in any of the settings?
Retrain YOLOv5 on a custom dataset
Please let me know if more information is needed.
Hi @wesley.chen,
It looks like you got a timeout before the compilation could converge. You can increase the timeout by adding the following command to the model script:
allocator_param(timeout=100h)
Let us know how it goes.
1 Like
Hi,
Thank you so much for replying to my post, and I apologize for the delay in getting back to you.
Regarding the issue I encountered earlier, I spent some time on it but was unable to reproduce the problem.
At that time, I decided to reinstall the Hailo Model Zoo, which reset some files (yaml, all) to their default state, and the subsequent compilation tasks went smoothly.
It’s possible that modifications to those files had caused the issue.
In any case, thank you very much for your help!
1 Like
Glad to hear things went well!
1 Like
