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?