I cannot train a detection model. I get the error: RuntimeError: Error in loading state_dict for SSD: Unexpected key(s) in state_dict:

Robotics & Edge Computing Jetson & Embedded Systems Jetson Nano
5

Hi,

This error might occur if net.to("cuda:0") is not working.

How do you install PyTorch for Jetson?
Could you try our package shared in the below topic?

Thanks.