Gpu training on host machine in jetson xavier agx module

I want to do gpu training with yolov5 on the host machine in jetson xavier agx module. After all the steps, I get this error after starting the training. I may be getting this error because my gpu memory is full, but I just reset xavier.

Hi,

Do you want to use Xavier as a host for training?
If yes, please check if your training sample does support the Jetson environment first.

Thanks.

how do i check this?

Hi,

You can contact the sample owner directly.
Thanks.