Jetson nano road following jetracer optimization error during model conversion to trt model

Hi, i am trying to optimise my resnet18 model using tensor2trt. But its failing with below error. Im using the existing road following repo
This is happening when i am trying to convert model to trt with torch.save(). I haven’t modified any code in the givin repo.

Github: jetracer/notebooks/optimize_model.ipynb at master · NVIDIA-AI-IOT/jetracer · GitHub

Cuda 10.2
Cudnn 8
Torch 1.6.0

Jetpack 4.5

Hi,

Do you use the same environment shared in the Jetracer?

https://github.com/NVIDIA-AI-IOT/jetracer/tree/master

6/01/2021 - JetPack 4.5.1 based image is updated. It is pre-configured for JetRacer. Detail here.

Thanks.

We followed the instructions provided in the github.!! What you mean by environment is it jetpack version?? Can you pls help me understand

Hi,

Yes, do you use the image shared in the link below?

Thanks.

Yes same image is used. It worked yesterday but i dono why it dint work for the past 5 weeks. So this happens only cos of the image? Or memory issue??

Hi,

Not sure but do you mean it works now?
Thanks.

Yes as you mentioned earlier about the image, i made sure to download it by myself and flash it. Another thing is i dint train the model as i had the pth file already… so i directly went for converting it into trt and it worked so not sure about the cause… will try to replicate and see

Thanks for the guidance

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.