custom-lib for img size 416

[thewantedlegend]

I compiled the code. But when I am trying to test model damoyolo_tinynasL20_Nl.onnx it exports an engine and gives error

terminate called after throwing an instance of 'thrust::system::system_error'
what(): trivial_device_copy D->H failed: cudaErrorIllegalAddress: an illegal memory access was encountered
Aborted (core dumped)

When I try with a model trained with size=640 it runs without any error.

[marcoslucianops]

Are you converting the pth file to onnx using the utils/export_damoyolo.py file?

[AkashgSeethos]

Are you able to solve this error?

[marcoslucianops]

I didn't get issues running here. Make sure you are following the doc: https://github.com/marcoslucianops/DeepStream-Yolo/blob/master/docs/DAMOYOLO.md