Converting YOLOX-s model to hef format

[thePrimeTux]

What changes should I make to convert the pretrained YOLOX-s model from YOLOX to hef format? I reused hailo_model_zoo/cfg/networks/yolox_s_wide_leaky.yaml and changed the end node names and then converted the model to hef format using the following command

hailomz compile --hw-arch hailo8l --yaml hailo_model_zoo/cfg/networks/yolox_s.yaml --ckpt yolox_s.onnx --calib-path hailo_model_zoo/coco_dataset --performance

But running hailortcli run yolox_s.hef gave the following error.

Running streaming inference (../yolox_s.hef):
Transform data: true
Type: auto
Quantized: true
[HailoRT] [error] CHECK failed - Failed to extract_detections, reg yolox_s_leaky/conv55_111 buffer_size should be 25600, but is 6400
[HailoRT] [error] CHECK_SUCCESS failed with status=HAILO_INVALID_ARGUMENT(2)
[HailoRT] [error] CHECK_SUCCESS failed with status=HAILO_INVALID_ARGUMENT(2)
[HailoRT] [error] CHECK_SUCCESS failed with status=HAILO_INVALID_ARGUMENT(2)
Network yolox_s_leaky/yolox_s_leaky: 100% | 0 | FPS: 0.00 | ETA: 00:00:00
[HailoRT CLI] [error] Failed waiting for threads with status HAILO_INVALID_ARGUMENT(2)
[HailoRT CLI] [error] CHECK_SUCCESS failed with status=HAILO_INVALID_ARGUMENT(2)
[HailoRT CLI] [error] CHECK_SUCCESS failed with status=HAILO_INVALID_ARGUMENT(2)
[HailoRT CLI] [error] CHECK_SUCCESS failed with status=HAILO_INVALID_ARGUMENT(2)
[HailoRT CLI] [error] CHECK_SUCCESS failed with status=HAILO_INVALID_ARGUMENT(2) - Error failed running inference